二、模型训练与优化(2):模型训练结果分析

目录

1. 训练过程中的指标

2. 测试集上的指标

3. 理解损失(loss)与准确率(accuracy)

3.1 损失(Loss)

3.2 准确率(Accuracy)

4. 如何分析训练曲线

4.1 训练准确率 vs. 验证准确率

4.2 训练损失 vs. 验证损失

5. 常见问题与应对方法

5.1 过拟合(Overfitting)

5.2 欠拟合(Underfitting)

5.3 学习率问题

6. 以 MNIST 为例的具体结果分析思路

6.1 训练和验证走势一致

6.2 训练准确率略高于验证准确率

6.3 测试集上的指标

7. 下一步要做什么

8. 实际操作中的一些建议

9. 小结


目前已经完成了模型的构建和训练,我们最终的目的是将AI模型部署到STM32上运行,所以需要对模型进行一定的处理。接下来对模型的结果进行分析以及对模型进行优化。

在模型完成训练之后,会得到若干评估指标(如损失、准确率),以及训练过程中产生的一系列数据(训练损失、训练准确率、验证损失、验证准确率等)。对这些数据进行分析,能够帮助我们理解模型的学习情况并判断下一步需要做什么优化。

以下是一个详细的结果分析流程,可以更好地理解和运用这些指标。


1. 训练过程中的指标

在 Keras 中,通过调用 model.fit(...) 方法进行模型训练时,默认会输出以下信息:

  1. Epoch:表示训练轮次,如 Epoch 1/5 表示第 1 轮,总共 5 轮。
  2. loss(训练损失):这是当前这轮训练结束时,模型在 训练数据 上的损失值。
  3. accuracy(训练准确率):表示模型在 训练数据 上的准确率(对于分类问题)。
  4. val_loss(验证损失):这是当前这轮训练结束时,模型在 验证数据(validation set)上的损失值。
  5. val_accuracy(验证准确率):表示模型在 验证数据 上的准确率。

为什么要区分训练和验证?

  • 训练集(train set):模型用来更新参数(学习)
  • 验证集(validation set):模型不会直接用来更新参数,而是用来监控训练过程,防止过拟合,并进行模型调参(如选择合适的轮数、学习率等)。

一个典型的训练输出示例如下:

Epoch 1/5
844/844 [==============================] - 4s 5ms/step - loss: 0.3208 - accuracy: 0.9109 - val_loss: 0.1426 - val_accuracy: 0.9595
Epoch 2/5
844/844 [==============================] - 4s 5ms/step - loss: 0.1462 - accuracy: 0.9573 - val_loss: 0.1097 - val_accuracy: 0.9683
Epoch 3/5
844/844 [==============================] - 4s 5ms/step - loss: 0.1019 - accuracy: 0.9701 - val_loss: 0.0878 - val_accuracy: 0.9748
...

从上面的输出可以看到,每个 epoch 结束后,会打印以下信息:

  • loss:本轮训练结束时的训练损失
  • accuracy:本轮训练结束时的训练准确率
  • val_loss:在验证集上测试的损失
  • val_accuracy:在验证集上测试的准确率

2. 测试集上的指标

训练结束后,通常会调用:

test_loss, test_acc = model.evaluate(x_test, y_test, verbose=2)
print("测试集上的准确率:", test_acc)
  • test_loss:最终在 测试集 上的损失
  • test_acc:最终在 测试集 上的准确率

为什么要单独使用测试集ÿ

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

魂兮-龙游

你的鼓励是我创作最大的动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值