ошибка (неверные параметры) из метода стратификации в sklearn.model_selection.train_test_split - PullRequest
0 голосов
/ 05 марта 2019

Я пытаюсь использовать метод стратификации в 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

Так что, пожалуйста, каждый может мне помочь.

1 Ответ

0 голосов
/ 06 марта 2019

Появившаяся ошибка указывает на то, что параметр statify не существует для этой функции.Не удивительно ... потому что есть опечатка ;-) это должно быть stratify вместо этого, и это должно работать с вашей версией scikit-learn.

...