ImportError: No module named model_selection

后端 未结 11 596
攒了一身酷
攒了一身酷 2021-01-30 08:16

I am trying to use train_test_split function and write:

from sklearn.model_selection import train_test_split

and this causes

11条回答
  •  抹茶落季
    2021-01-30 08:43

    I guess you have the wrong version of scikit-learn, a similar situation was described here on GitHub. Previously (before v0.18), train_test_split was located in the cross_validation module:

    from sklearn.cross_validation import train_test_split
    

    However, now it's in the model_selection module:

    from sklearn.model_selection import train_test_split
    

    so you'll need the newest version.

    To upgrade to at least version 0.18, do:

    pip install -U scikit-learn
    

    (Or pip3, depending on your version of Python). If you've installed it in a different way, make sure you use another method to update, for example when using Anaconda.

提交回复
热议问题