Skip to content

Commit 62ec7c2

Browse files
chore: Replace cross_validation with CrossValidationReport in warnings (probabl-ai#1601)
the `cross_validation` function doesn't exist anymore. It was replace in PR probabl-ai#847 with the `CrossValidationReport` class. Please let me know if this change LGTY-- or should I've replaced "skore's `cross_validation` function." with something in sklearn (e.g. - this page: https://scikit-learn.org/stable/modules/cross_validation.html#cross-validation)? Thank you :)
1 parent 289b5f3 commit 62ec7c2

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

skore/src/skore/sklearn/train_test_split/warning/high_class_imbalance_too_few_examples_warning.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,8 @@ class HighClassImbalanceTooFewExamplesWarning(TrainTestSplitWarning):
2525
"fewer than 100 examples in the test set. In this case, using train_test_split "
2626
"may not be a good idea because of high variability in the scores obtained on "
2727
"the test set. We suggest three options to tackle this challenge: you can "
28-
"increase test_size, collect more data, or use skore's cross_validate function."
28+
"increase test_size, collect more data, or use skore's CrossValidationReport "
29+
"with the `cv_splitter` parameter of your choice. "
2930
)
3031

3132
@staticmethod

skore/src/skore/sklearn/train_test_split/warning/high_class_imbalance_warning.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@ class HighClassImbalanceWarning(TrainTestSplitWarning):
2626
"case, using train_test_split may not be a good idea because of high "
2727
"variability in the scores obtained on the test set. "
2828
"To tackle this challenge we suggest to use skore's "
29-
"cross_validate function."
29+
"CrossValidationReport with the `cv_splitter` parameter "
30+
"of your choice. "
3031
)
3132

3233
@staticmethod

0 commit comments

Comments
 (0)