AttributeError:模块“”sklearn“”没有属性“”model_selection“”
AttributeError:模块“”sklearn“”没有属性“”model_selection“”
提问于 2018-12-12 19:51:53
当我想使用sklearn.model_selection.train_test_split来拆分训练集和测试集时，它会抛出如下错误：
AttributeError: module 'sklearn' has no attribute 'model_selection'我的代码如下：
import pandas as pd
import sklearn
data = pd.read_csv('SpeedVideoDataforModeling.csv',header=0,)
data_x = data.iloc[:,1:4]
data_y = data.iloc[:,4:]
x_train , x_test, y_train, y_test =
sklearn.model_selection.train_test_split(data_x,data_y,test_size = 0.2)在Pycharm中，scikit-learn包版本为0.19.1。enter image description here感谢您的帮助！
回答 2
Stack Overflow用户
发布于 2018-12-12 19:54:30
你需要
import sklearn.model_selection在调用函数之前
Stack Overflow用户
发布于 2021-02-15 04:03:06
您可以像from sklearn.model_selection import train_test_split一样导入。官方文档中的一个示例:)
>>> import numpy as np
>>> from sklearn.model_selection import train_test_split
>>> X, y = np.arange(10).reshape((5, 2)), range(5)
>>> X
array([[0, 1],
[2, 3],
[4, 5],
[6, 7],
[8, 9]])
>>> list(y)
[0, 1, 2, 3, 4]
>>> X_train, X_test, y_train, y_test = train_test_split(
... X, y, test_size=0.33, random_state=42)
...
>>> X_train
array([[4, 5],
[0, 1],
[6, 7]])
>>> y_train
[2, 0, 3]
>>> X_test
array([[2, 3],
[8, 9]])
>>> y_test
[1, 4]页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/53742441
复制相关文章
相似问题
腾讯云开发者
