TypeError： fit() 得到一个意外的关键字参数'test'

Question

model = Sequential()
model.add(LSTM(512, activation='relu', return_sequences=True))
model.add(LSTM(512, activation='relu'))
model.add(Dense(2))
model.compile(loss='mse', optimizer='adam', metrics='accuracy')
model.fit(X_train, y_train, test=(x_test, y_test), validation=(x_val, y_val), epochs=50)

错误如下：

TypeError: fit() got an unexpected keyword argument 'test'

当我删除测试时，我会得到相同的验证错误，如果我删除两者，我会收到这个错误：

TypeError: fit() got multiple values for argument 'epochs'

我正在尝试让LSTM与训练、测试和验证集一起运行。

Answer 1

您不将测试数据集传递给fit()，而是在拟合期间使用验证集，在通过调用model.evaluate:评估训练模型时使用测试集

model.fit(X_train, y_train, validation_data=(x_val, y_val), epochs=50)

用于评估：

results = model.evaluate(x_test, y_test, batch_size= #add your batch size here)

参考：https://www.tensorflow.org/guide/keras/train_and_evaluate