StratifiedKFold in Python geben Fehler

Question

in Python I

>>> from sklearn.model_selection import StratifiedKFold 

>>> cv = StratifiedKFold(y, n_folds=5) 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
TypeError: __init__() got an unexpected keyword argument 'n_folds' 

>>> print('The sklearn version is {}.'.format(sklearn.__version__)) 
The sklearn version is 0.18.1. 

I 0.18.1 haben sklearn bin importieren. Irgendeine Idee, warum ich diesen Fehler bekomme?

Answer 1

Der Parameter für die Anzahl der Falten ist n_splits, nicht n_folds

See: http://scikit-learn.org/stable/modules/generated/sklearn.model_selection.StratifiedKFold.html