Scipy優化fmin ValueError：使用序列設置數組元素

Question

當使用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.

如果有任何幫助可以解釋我哪里出錯了，我將不勝感激。

Answer 1

原因是你給fmin的起點（initial_theta）不是一維數組而是二維數組。 因此，在第二次迭代中，fmin傳遞一維數組（這就是它應該如何工作），結果變為非標量。

所以你應該重構你的成本函數來接受1d數組作為第一個參數。

最簡單的改變是使代碼工作是在傳遞給fmin之前將initial_theta展平，並在cost_function中將theta重塑為（X.shape [1]，1），如果你願意的話。

Answer 2

cost_function應該返回一個標量，但是你的返回值J是某種數組。