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



我遇到了一个奇怪的问题,一些线性回归的东西从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.

这里的例子运行得很好。extractFeaturesFromData返回的X和y类型为'list',与示例中相同。如果我使用示例页面中的虚拟X和y,它可以正常工作,但使用我的X和y会导致抛出错误。

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

明白了!这与我的代码本身完全无关;我导入的其中一个文件比其他文件大一些,并且(我认为)被自动分割成一个数组,导致错误。删除该文件后,一切正常运行。

相关内容

  • 没有找到相关文章