8000 Update the model_selection module with changes from master · raghavrv/scikit-learn@f9bfe39 · GitHub
[go: up one dir, main page]
Skip to content

Commit f9bfe39

Browse files
raghavrvraghavrv
committed
Update the model_selection module with changes from master
From scikit-learn#5161 - MAINT remove redundant p variable - Add check for sparse prediction in cross_val_predict From scikit-learn#5201 - DOC improve random_state param doc From scikit-learn#5190 - LabelKFold and test From scikit-learn#4583 - LabelShuffleSplit and tests
1 parent de2c8a2 commit f9bfe39

File tree

5 files changed

+396
-73
lines changed

5 files changed

+396
-73
lines changed

‎sklearn/model_selection/__init__.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
11
from ._split import BaseCrossValidator
22
from ._split import KFold
3+
from ._split import LabelKFold
34
from ._split import StratifiedKFold
45
from ._split import LeaveOneLabelOut
56
from ._split import LeaveOneOut
67
from ._split import LeavePLabelOut
78
from ._split import LeavePOut
89
from ._split import ShuffleSplit
10+
from ._split import LabelShuffleSplit
911
from ._split import StratifiedShuffleSplit
1012
from ._split import PredefinedSplit
1113
from ._split import train_test_split
@@ -23,10 +25,11 @@
2325
from ._search import ParameterSampler
2426
from ._search import fit_grid_point
2527

26-
__all__ = ('BaseCrossValidator', 'GridSearchCV', 'KFold', 'LeaveOneLabelOut',
27-
'LeaveOneOut', 'LeavePLabelOut', 'LeavePOut', 'ParameterGrid',
28-
'ParameterSampler', 'PredefinedSplit', 'RandomizedSearchCV',
29-
'ShuffleSplit', 'StratifiedKFold', 'StratifiedShuffleSplit',
30-
'check_cv', 'cross_val_predict', 'cross_val_score',
31-
'fit_grid_point', 'learning_curve', 'permutation_test_score',
32-
'train_test_split', 'validation_curve')
28+
__all__ = ('BaseCrossValidator', 'GridSearchCV', 'KFold', 'LabelKFold',
29+
'LeaveOneLabelOut', 'LeaveOneOut', 'LeavePLabelOut', 'LeavePOut',
30+
'ParameterGrid', 'ParameterSampler', 'PredefinedSplit',
31+
'RandomizedSearchCV', 'ShuffleSplit', 'LabelShuffleSplit',
32+
'StratifiedKFold', 'StratifiedShuffleSplit', 'check_cv',
33+
'cross_val_predict', 'cross_val_score', 'fit_grid_point',
34+
'learning_curve', 'permutation_test_score', 'train_test_split',
35+
'validation_curve')

0 commit comments

Comments
 (0)
0