StratifiedKFold在Python給錯誤

Question

在python我進口

>>> 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. 

我有sklearn 0.18.1。任何想法，爲什麼我得到這個錯誤？

Answer 1

爲褶皺的數目的參數是n_splits，不n_folds

參見： http://scikit-learn.org/stable/modules/generated/sklearn.model_selection.StratifiedKFold.html