Scipy优化fmin ValueError:设置具有序列的数组元素
问题内容:
使用scipy.optimize
s时fmin
出现错误,我不明白:
ValueError: setting an array element with a sequence.
这是一个简单的平方误差示例,以进行演示:
import numpy as np
from scipy.optimize import fmin
def cost_function(theta, X, y):
m = X.shape[0]
error = X.dot(theta) - y
J = 1/(2*m) * error.T.dot(error)
return J
X = np.array([[1., 1.],
[1., 2.],
[1., 3.],
[1., 4.]])
y = np.array([[2],[4],[6],[8]])
initial_theta = np.ones((X.shape[1], 1)) * 0.01
# test cost_function
print cost_function(initial_theta, X, y)
# [[ 14.800675]] seems okay...
# but then error here...
theta = fmin(cost_function, initial_theta, args=(X, y))
#Traceback (most recent call last):
# File "C:\Users\me\test.py", line 21, in <module>
# theta = fmin(cost_function, initial_theta, args=(X, y))
# File "C:\Python27\lib\site-packages\scipy\optimize\optimize.py", line 278, in fmin
# fsim[0] = func(x0)
#ValueError: setting an array element with a sequence.
我将不胜感激,希望能为您解释我在哪里出错了。
问题答案:
原因是您给fmin的起点(initial_theta)不是1D数组,而是2D数组。因此,在第二次迭代中,fmin传递了一个1D数组(这就是它应该起作用的方式),结果变为非标量。
因此,您应该重构成本函数以接受1d数组作为第一个参数。
最简单的更改是使代码正常工作是在传递给fmin之前先将initial_theta展平,然后根据需要将cost_function中的theta重塑为(X.shape
[1],1)。