python - sklearn.model_selection.train_test_split 中分层方法的（无效参数）错误

Question

我正在尝试在sklearn.model_selection.train_test_split. 这是我的代码：

x=df["x"]
y=df["y"]
X_train, X_test, Y_train, Y_test = train_test_split( x, y, test_size=0.2, random_state=42,statify= y)

但我得到这个错误

Invalid parameters passed: {'statify': 386 real....Name: y, Length: 527, dtype: object}

我在寻找这个答案，他们提到这stratify splitting是 0.17 版中的新内容，我必须更新我的sklearn. 我找了我的版本。这是0.20.2

scikit-learn              0.20.2                   pypi_0    pypi

所以请任何人都可以帮助我。

score 2 · Accepted Answer

您遇到的错误表明statify此函数的参数不存在。难怪......因为有一个错字;-) 它应该是stratify相反的，这应该适用于你的scikit-learn.

score 0 · Accepted Answer

此错误是由于拼写错误 'statify' 造成的，请将其更改为 'stratify' 它将起作用。请检查此链接sklearn

`

2 回答 2