I am trying linear regression from a data but when I try the following:

from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression

it gives me this error:

line 4, in <module>
    from sklearn.model_selection import train_test_split
KeyError: 32

sklearn, numpy and scipy are all installed but then also I am not able to use sklearn.model_selection and linear_model.

How can I solve this problem? I have already tried upgrading every module.

1 Answers

3
Ralf On

What version of sklearn are you using?

In versions 0.19 and prior, the function train_test_split() was located in sklearn.cross_validation. To use it:

from sklearn.cross_validation import train_test_split

From version 0.20 onwards it can be found in sklearn.model_selection To use it:

from sklearn.model_selection import train_test_split

Does that help?