Scipy优化fmin ValueError：使用序列设置数组元素答案

【问题标题】：Scipy optimize fmin ValueError: setting an array element with a sequenceScipy优化fmin ValueError：使用序列设置数组元素
【发布时间】：2012-04-04 12:22:12
【问题描述】：

当使用scipy.optimize 的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 需要一个值。要么返回J[0,0]，要么将成本函数重写为J = 1/(2*m) * np.sum(error**2)

标签： python numpy scipy

【解决方案1】：

原因是你给 fmin 的起点（initial_theta）不是一维数组而是二维数组。所以在第二次迭代中， fmin 传递了一个一维数组（这就是它应该如何工作的），结果变成了非标量。

因此，您应该重构成本函数以接受一维数组作为第一个参数。

最简单的更改是使代码工作是在传递给 fmin 之前展平 initial_theta 并将 cost_function 内的 theta 重新整形为 (X.shape[1],1) 如果您愿意。

【讨论】：

感谢您的解释。是的，维度的数量让我感到困惑。所以，我现在最初将 theta 和 y 都设置为一个暗度，成本函数返回值作为一个标量计算，无需任何进一步的调整。干杯。

【解决方案2】：

cost_function 应该返回一个标量，但您的返回值 J 是某种数组。

【讨论】：