search

squaredev-io / whitebox

[Not Actively Maintained] Whitebox is an open source E2E ML monitoring platform with edge capabilities that plays nicely with kubernetes
https://squaredev.io/whitebox/
MIT License
184 stars 5 forks source link

Update pipelines.py #157

Open Mikehade opened 1 year ago

Mikehade commented 1 year ago

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

Mikehade commented 1 year ago

Stratify parameter added added

momegas commented 1 year ago

Some tests are failing 🥶

Mikehade commented 1 year ago

I will check out the errors to find a solution

Mikehade commented 1 year ago

I have corrected the stratify variable