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