проблемы с функцией 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: Фигуры должны быть одинакового ранга, но имеют 2 и 0 для «Назначить» (op: «Назначить») с входными фигурами: [1,1], [].