函数set_weights（）的问题

Question

我要在模型上安装一系列砝码。不断地发现一个相同的错误。请告诉我如何正确进行操作。

我只设法减轻体重。尽管这样做可以，但是比例尺的数量可能会增加，并且不断更改代码以解决此问题将不方便。

model.set_weights([array([[x[number][0]]], dtype=float32), array([x[number][1]], dtype=float32)])

def baseline_model():
    model = Sequential()
    model.add(Dense(1, activation='linear', input_dim=1))
    model.compile(optimizer=optimizers.RMSprop(lr=0.001), loss='mean_squared_error')
    return model


def fitness(x):
    model = baseline_model()
    model.set_weights(x)
    y_predict = model.predict(data)
    return K.eval(losses.mean_squared_error(y, np.ravel(y_predict)))


x = [0.5, 0.6]
fitness(x)

  

ValueError：形状必须相等，但对于输入形状为[1,1]，[]的“分配”（操作：“分配”），形状必须为2和0。