scikit中的新更新学习交叉验证库

Question

我正在从udemy的机器学习课程a-z中学习ML。在数据预处理的代码模板中，其表示为

from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 0)

但是当我运行它时，它没有sklearn.cross_validation。在谷歌搜索中，我发现我们应该使用sklearn。如何在最新版本的sklearn.model_selection.train_test_split中做到这一点，但这不是一个类。如何在最新版本中做到这一点

Answer 1

请使用

from sklearn.model_selection import train_test_split

使用sklearn 0.18及更高版本。

Answer 2

train_test_split已从scikit-learn的0.18中的cross_validation移至model_selection。请将scikit-learn更新为0.18或最新版本。

您可以通过以下方式通过命令行进行更新：

pip install --upgrade scikit-learn

或

conda update scikit-learn   (if using conda environment).

---