whitebox icon indicating copy to clipboard operation
whitebox copied to clipboard
verified publisher icon squaredev-io

Update pipelines.py

Open Mikehade opened this issue 2 years ago • 4 comments

Update pipelines.py by adding stratify parameter to train_test_split function of the sklearn library

  • [x] closes #91 (Replace 91 with the GitHub issue number)
  • [x] Tests added and passed if fixing a bug or adding a new feature.
  • [x] All code checks passed.
  • [ ] Added type annotations to new arguments/methods/functions.
  • [ ] Updated documentation.

Mikehade avatar Mar 15 '23 16:03 Mikehade

Stratify parameter added added

Mikehade avatar Mar 16 '23 22:03 Mikehade

Some tests are failing 🥶

momegas avatar Mar 20 '23 11:03 momegas

I will check out the errors to find a solution

Mikehade avatar Mar 21 '23 07:03 Mikehade

I have corrected the stratify variable

Mikehade avatar Mar 23 '23 21:03 Mikehade