feat: Add `HighClassImbalanceTooFewExamples` warning to `train_test_split`

probabl-ai / skore

Skore lets you "Own Your Data Science." It provides a user-friendly interface to track and visualize your modeling results, and perform evaluation of your machine learning models with scikit-learn.

https://probabl-ai.github.io/skore/

MIT License

76 stars 7 forks source link

feat: Add `HighClassImbalanceTooFewExamples` warning to `train_test_split` #772

Closed augustebaum closed 5 days ago

augustebaum commented 6 days ago

Export train_test_split warnings through train_test_split.warning module
refactor tests to mirror src
Add new warning

Would welcome suggestions for a shorter name ^^'

Addresses #684