Python sklearn linear_model对象在传递列表时抛出ValueError

时间:2016-10-07 15:07:11

标签: python-2.7 scikit-learn

我遇到一个奇怪的问题,来自sklearn的一些线性回归。具体来说,是linear_model。

我正在尝试进行一些基本的机器学习,因此我的脚本中有一部分可以梳理我的数据并将功能提取到列表(列表)X中,然后将另一部分提供给适合的功能。所以我(大致)

from sklearn import linear_model
X, y = extractFeaturesFromData(data,numfeatures)   # my homemade function
reg = linear_model.LinearRegression()
reg.fit(X,y)

当我运行时,我得到(以及回溯)

ValueError: setting an array element with a sequence.

示例here运行正常。并且extractFeaturesFromData返回的X和y的类型为“list”,与示例中的相同。如果我使用示例页面中的虚拟X和y,它工作正常,但使用我的会导致它抛出错误。

我尝试改变提取到X中的特征数量,并打印出从我的函数返回的X和y(这表明它们与示例中的虚拟对应格式相同),但还没有到目前为止运气好我在运行10.9.5的macbook上运行python 2.7。知道为什么会这样吗?任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:0)

想出来!这与我的代码本身完全无关;我导入的其中一个文件比其他文件大一点,而且(我认为)会自动拆分成一个数组,从而导致错误。删除该文件使一切运行正常。