Scipy optimize fmin ValueError: установка элемента массива с последовательностью

При использовании 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.

Буду благодарен за любую помощь, чтобы объяснить, где я ошибаюсь.

Ответ 1

Причина в том, что начальная точка (initial_theta), которую вы дали fmin, не является массивом 1D, а 2D-массивом. Таким образом, на второй итерации fmin передает 1D-массив (как он должен работать), и результат становится нескалярным.

Итак, вы должны реорганизовать свою функцию стоимости, чтобы принять 1d массивы в качестве первого аргумента.

Самое простое изменение заключается в том, чтобы заставить работать код, чтобы сгладить начальную_тету перед переходом к fmin и изменить форму theta inside cost_function на (X.shape [1], 1), если хотите.

Ответ 2

cost_function должен возвращать скаляр, но ваше возвращаемое значение J является массивом какого-либо типа.