提前停止(Early Stopping)
观察验证损失的变化,比较提前停止与固定训练轮数。
如果验证损失连续若干轮没有至少降低“最小改进量”,训练就停止。程序深复制验证损失最小时的 state_dict,并在训练结束后恢复这组参数;因此“最佳轮”与“实际停止轮”可能不同。所有轮次都从 1 开始记。
验证集 ROC 曲线
训练损失、验证损失与停止轮次
训练集与验证集指标
最终测试集评价
运行状态