Reputation: 511
I am trying to split my dataset into training and testing dataset, but I am getting this error:
X_train,X_test,Y_train,Y_test = sklearn.cross_validation.train_test_split(X,df1['ENTRIESn_hourly'])
AttributeError Traceback (most recent call last)
<ipython-input-53-5445dab94861> in <module>()
----> 1 X_train,X_test,Y_train,Y_test = sklearn.cross_validation.train_test_split(X,df1['ENTRIESn_hourly'])
AttributeError: module 'sklearn' has no attribute 'cross_validation'
How can I handle this?
Upvotes: 50
Views: 81022
Reputation: 1
Thanks! Successful with this in Colab:
from sklearn.model_selection import train_test_split
Upvotes: 0
Reputation: 2022
sklearn
does not automatically import its subpackages. If you only imported via: import sklearn
, then it won't work. Import with import sklearn.cross_validation
instead.
Further, sklearn.cross_validation
will be deprecated in version 0.20. Use sklearn.model_selection.train_test_split
instead.
Upvotes: 130
Reputation: 258
"cross_validation" name is now deprecated and was replaced by "model_selection" inside the new anaconda versions. So you can use
from sklearn.model_selection import train_test_split
Upvotes: 2
Reputation: 2080
The equivalent to cross_validation in sklearn is:
sklearn.model_selection
Upvotes: 2
Reputation: 10058
Try this:
from sklearn.model_selection import train_test_split
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.33, random_state=101)
Upvotes: 8
Reputation: 79
you can try this
X_train,X_test,Y_train,Y_test =
sklearn.model_selection.train_test_split(X,boston_df.price)
Upvotes: 2