Source code for evalml.preprocessing.data_splitters.sk_splitters
"""SKLearn data splitter wrapper classes."""
from sklearn.model_selection import KFold, StratifiedKFold
[docs]class KFold(KFold):
"""Wrapper class for sklearn's KFold splitter."""
@property
def is_cv(self):
"""Returns whether or not the data splitter is a cross-validation data splitter.
Returns:
bool: If the splitter is a cross-validation data splitter
"""
return True
[docs]class StratifiedKFold(StratifiedKFold):
"""Wrapper class for sklearn's Stratified KFold splitter."""
@property
def is_cv(self):
"""Returns whether or not the data splitter is a cross-validation data splitter.
Returns:
bool: If the splitter is a cross-validation data splitter
"""
return True