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

Question

我正在 udemy 上通过机器学习课程 az 学习机器学习。在数据预处理的代码模板中给出为

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

在 scikit-learn 的 0.18 版本中，train_test_split 从 cross_validation 移到了 model_selection。请将 scikit-learn 更新到 0.18 或最新版本。您可以通过以下方式通过命令行更新：pip install --upgrade scikit-learn或者conda update scikit-learn   (if using conda environment).