python – 来自sympy的scipy优化

2023年4月9日 285次阅读

我有四个用Sympy象征性计算的函数然后lambdified：

deriv_log_s_1 = sym.lambdify((z, m_1, m_2, s_1, s_2), deriv_log_sym_s_1, modules=['numpy', 'sympy'])
deriv_log_s_2 = sym.lambdify((z, m_1, m_2, s_1, s_2), deriv_log_sym_s_2, modules=['numpy', 'sympy'])
deriv_log_m_1 = sym.lambdify((z, m_1, m_2, s_1, s_2), deriv_log_sym_m_1, modules=['numpy', 'sympy'])
deriv_log_m_2 = sym.lambdify((z, m_1, m_2, s_1, s_2), deriv_log_sym_m_2, modules=['numpy', 'sympy'])

从这些函数中,我定义了一个优化的成本函数：

def cost_function(x, *args):

    m_1, m_2, s_1, s_2 = x     

    print(args[0])    

    T1 = np.sum([deriv_log_m_1(y, m_1, m_2, s_1, s_2) for y in args[0]])   
    T2 = np.sum([deriv_log_m_2(y, m_1, m_2, s_1, s_2) for y in args[0]]) 


    T3 = np.sum([deriv_log_m_1(y, m_1, m_2, s_1, s_2) for y in args[0]])   
    T4 = np.sum([deriv_log_m_1(y, m_1, m_2, s_1, s_2) for y in args[0]])   

    return T1 + T2 + T3 + T4

我的函数cost_function按预期工作：

a = 48.7161
b = 16.3156
c = 17.0882
d = 7.0556
z = [0.5, 1, 2, 1.2, 3]

test = cost_function(np.array([a, b, c, d]).astype(np.float32), z)

但是,当我尝试优化它时：

from scipy.optimize import fmin_powell

res = fmin_powell(cost_function, x0=np.array([a, b, c, d], dtype=np.float32), args=(z, ))

它引发了以下错误：

AttributeError: 'Float' object has no attribute 'sqrt'

我不明白为什么会出现这样的错误,因为我的cost_function本身并没有引起任何错误.

最佳答案 解决方案是,并且我不知道为什么,将输入转换为numpy.float：

m_1 = np.float32(m_1)
m_2 = np.float32(m_2)
s_1 = np.float32(s_1)
s_2 = np.float32(s_2)
点赞