python - 如何将2个矩阵放入scipy.optimize.minimize？

我使用scipy.optimize.minimize函数。
我的目的是获取w,z以使f(w,z)最小化
w和z均为n×m矩阵:

[[1,1,1,1],
 [2,2,2,2]]

f(w，z)是接收参数w和z。

我已经尝试过以下表格:

def f(x):
   w = x[0]
   z = x[1]
   ...

minimize(f, [w,z])

但是，最小化效果不佳。

将两个矩阵(n×m)放入scipy.optimize.minimize的有效形式是什么？

最佳答案

Optimize需要一维矢量来进行优化。您走在正确的轨道上。您需要将参数放平到minimize，然后在f中以x = np.reshape(x, (2, m, n))开头，然后拉出w和z，您应该有业务了。

我以前遇到过这个问题。例如，同时优化多个不同类别中的矢量部分。通常，我通常使用一个将事物映射到一维向量的函数，然后使用另一个将数据拉回到对象中的函数，以便评估成本函数。如:

def toVector(w, z):
    assert w.shape == (2, 4)
    assert z.shape == (2, 4)
    return np.hstack([w.flatten(), z.flatten()])

def toWZ(vec):
    assert vec.shape == (2*2*4,)
    return vec[:2*4].reshape(2,4), vec[2*4:].reshape(2,4)

def doOptimization(f_of_w_z, w0, z0):
    def f(x):
        w, z = toWZ(x)
        return f_of_w_z(w, z)

    result = minimize(f, toVec(w0, z0))
    # Different optimize functions return their
    # vector result differently. In this case it's result.x:
    result.x = toWZ(result.x)
    return result

关于python - 如何将2个矩阵放入scipy.optimize.minimize？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/31292374/