Я пытаюсь использовать метод стратификации в 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
Так что, пожалуйста, каждый может мне помочь.