在 README 中直接展示评估与训练曲线图片
This commit is contained in:
parent
44c70250e3
commit
76b56dd64b
1 changed files with 13 additions and 14 deletions
27
README.md
27
README.md
|
|
@ -184,34 +184,33 @@ trash-division/
|
|||
|
||||
在验证集上推理,生成三张评估图表:
|
||||
|
||||
| 图表 | 说明 |
|
||||
|---|---|
|
||||
| `confusion_matrix.png` | 混淆矩阵,展示各类别的预测分布 |
|
||||
| `roc_curve.png` | ROC 曲线(One-vs-Rest),含各类别 AUC 和 Macro-avg AUC |
|
||||
| `pr_curve.png` | Precision-Recall 曲线,含各类别 Average Precision |
|
||||
|
||||
```bash
|
||||
python Evaluate.py
|
||||
```
|
||||
|
||||
脚本顶部的 `MODEL_PATH`、`DATA_ROOT`、`BATCH_SIZE`、`NUM_WORKERS` 可按需修改。
|
||||
|
||||
**混淆矩阵**
|
||||
|
||||

|
||||
|
||||
**ROC 曲线**
|
||||
|
||||

|
||||
|
||||
**PR 曲线**
|
||||
|
||||

|
||||
|
||||
### Curve.py — 训练曲线
|
||||
|
||||
从 `training_log.csv` 读取训练日志,绘制四张子图:
|
||||
|
||||
| 子图 | 说明 |
|
||||
|---|---|
|
||||
| Loss vs Epoch | 训练/验证损失曲线 |
|
||||
| F1 Score vs Epoch | 训练/验证宏平均 F1 曲线 |
|
||||
| Accuracy vs Epoch | 训练/验证准确率曲线 |
|
||||
| Learning Rate vs Epoch | 余弦退火学习率变化曲线 |
|
||||
|
||||
```bash
|
||||
python Curve.py
|
||||
```
|
||||
|
||||
输出文件为 `training_curves.png`。
|
||||

|
||||
|
||||
## 许可证
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue