我目前正在研究一个高度不平衡的多类分类问题。我想保存我的模型权重以获得最佳时期,但我对应该选择哪个指标感到困惑?
这是我的训练进度条:
我使用ModelCheckpoint
回调tf.keras
和监控val_loss
作为保存最佳模型权重的指标。
正如你在图片中看到的,
- 在第 8 个 epoch,我得到了一个
val_acc = 0.9845
butval_loss = 0.629
和准确率和召回率也很高。 - 但是在第三个时代我得到了
val_acc = 0.9840
但是val_loss = 0.590
我知道差异并不大,但在这种情况下,相信不平衡数据集的理想指标是什么?