model.fit(x_train, y_train, epochs=20, validation_data=(x_test, y_test))如何提取该行代码中产生的准确率
时间: 2023-12-24 07:47:05 浏览: 81
浅谈keras通过model.fit_generator训练模型(节省内存)
在使用 `model.fit()` 训练模型时,可以指定 `metrics` 参数来设置要计算的指标。如果要计算准确率,可以将 `metrics` 参数设置为 `['accuracy']`,如下所示:
```
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
model.fit(x_train, y_train, epochs=20, validation_data=(x_test, y_test))
```
在训练完成后,可以通过 `model.history` 属性来获取训练过程中的指标值。其中,训练集和验证集的准确率分别存储在 `acc` 和 `val_acc` 中。可以通过以下代码来获取最后一个 epoch 的验证集准确率:
```
val_acc = model.history.history['val_acc'][-1]
```
如果要获取训练集准确率,可以将 `val_acc` 改为 `acc`。
阅读全文