From 76b56dd64b41e12e4917b420d00c436e773fa2f1 Mon Sep 17 00:00:00 2001 From: yukun-hh Date: Thu, 14 May 2026 16:04:31 +0800 Subject: [PATCH] =?UTF-8?q?=E5=9C=A8=20README=20=E4=B8=AD=E7=9B=B4?= =?UTF-8?q?=E6=8E=A5=E5=B1=95=E7=A4=BA=E8=AF=84=E4=BC=B0=E4=B8=8E=E8=AE=AD?= =?UTF-8?q?=E7=BB=83=E6=9B=B2=E7=BA=BF=E5=9B=BE=E7=89=87?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 27 +++++++++++++-------------- 1 file changed, 13 insertions(+), 14 deletions(-) diff --git a/README.md b/README.md index a34e646..0724ca2 100644 --- a/README.md +++ b/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` 可按需修改。 +**混淆矩阵** + +![confusion_matrix](confusion_matrix.png) + +**ROC 曲线** + +![roc_curve](roc_curve.png) + +**PR 曲线** + +![pr_curve](pr_curve.png) + ### 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`。 +![training_curves](training_curves.png) ## 许可证