__init __（）得到了意外的关键字参数'k_folds'

Question

我正在使用Python 3.6.3版本和sklearn.__version__ '0.20.3' 我遇到__INIT__

的问题

#cv = KFold(len(X), k_folds=6, shuffle=True)
from sklearn.model_selection import StratifiedKFold
cv = StratifiedKFold(len(X), k_folds=6, shuffle=True)
results = []
y_pred_list = []

__init__() got an unexpected keyword argument 'k_folds'

Answer 1

根据the docs，只有3个可接受的参数：

• n_splits：int
• shuffle：bool
• random_state：int

您是要使用n_splits而不是k_folds吗？

Answer 2

class sklearn.model_selection.StratifiedKFold(n_splits=’warn’, shuffle=False, random_state=None)

此类中没有k_folds这样的参数