python - 导入错误 - 无法从 'train_test_split' 导入名称 'sklearn.preprocessing'

Question

我想导入 train_test_split 以将我的数据集拆分为测试数据集和训练数据集，但发生了导入错误。

我尝试了所有这些，但没有一个有效:

conda upgrade scikit-learn
pip uninstall scipy
pip3 install scipy

pip uninstall sklearn
pip uninstall scikit-learn
pip install sklearn

这是产生错误的代码:

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

这里是错误:

from sklearn.preprocessing import train_test_split
Traceback (most recent call last):

  File "<ipython-input-3-e25c97b1e6d9>", line 1, in <module>
    from sklearn.preprocessing import train_test_split

ImportError: cannot import name 'train_test_split' from 'sklearn.preprocessing' (C:\ProgramData\Anaconda3\lib\site-packages\sklearn\preprocessing\__init__.py)

Answer 1

train_test_split 不在 preprocessing 中，它在 model_selection 和 cross_validation 中，所以你的意思是:

from sklearn.model_selection import train_test_split

或者:

from sklearn.cross_validation import train_test_split