@@ -223,12 +223,12 @@ def __len__(self):
223
223
/ factorial (self .p ))
224
224
225
225
226
- class BaseKFold (with_metaclass (ABCMeta , _PartitionIterator )):
226
+ class _BaseKFold (with_metaclass (ABCMeta , _PartitionIterator )):
227
227
"""Base class to validate KFold approaches"""
228
228
229
229
@abstractmethod
230
230
def __init__ (self , n , n_folds , indices , k = None ):
231
- super (BaseKFold , self ).__init__ (n , indices )
231
+ super (_BaseKFold , self ).__init__ (n , indices )
232
232
if k is not None : # pragma: no cover
233
233
warnings .warn ("The parameter k was renamed to n_folds and will be"
234
234
" removed in 0.15." , DeprecationWarning )
@@ -246,7 +246,7 @@ def __init__(self, n, n_folds, indices, k=None):
246
246
% (self .n_folds , self .n ))
247
247
248
248
249
- class KFold (BaseKFold ):
249
+ class KFold (_BaseKFold ):
250
250
"""K-Folds cross validation iterator.
251
251
252
252
Provides train/test indices to split data in train test sets. Split
@@ -334,7 +334,7 @@ def __len__(self):
334
334
return self .n_folds
335
335
336
336
337
- class StratifiedKFold (BaseKFold ):
337
+ class StratifiedKFold (_BaseKFold ):
338
338
"""Stratified K-Folds cross validation iterator
339
339
340
340
Provides train/test indices to split data in train test sets.
0 commit comments