摘要: 早停法:防止过拟合,当验证集 loss 多轮不再下降时,提前停止训练。 early_stopping.py class EarlyStopping: def __init__(self, patience=3, min_delta=0): """ 参数: - patience: 忍耐轮数(如果验证集 阅读全文
posted @ 2025-07-25 16:23 daviyoung 阅读(169) 评论(0) 推荐(0)
摘要: 绘制训练/测试的 Loss 和 Accuracy 曲线,直观判断模型训练效果是否良好。 图像解释 Loss 曲线(损失函数) 反映模型在训练和测试过程中的错误程度,越低越好。 Accuracy 曲线(准确率) 反映模型预测正确的比例,越高越好。 你可以从图中看到什么? 1. 是否在收敛? Loss 阅读全文
posted @ 2025-07-25 14:49 daviyoung 阅读(85) 评论(0) 推荐(0)
摘要: 模型: neural_network_model.py from torch import nn # 定义神经网络模型 class NeuralNetwork(nn.Module): def __init__(self): super().__init__() self.flatten = nn.F 阅读全文
posted @ 2025-07-25 10:08 daviyoung 阅读(38) 评论(0) 推荐(0)