PyTorch 文本分类系统（含 PyQt5 图形界面）

基于 Transformer 的自动作文评分 / 文本分类实验，包含训练脚本与一个 PyQt5 桌面演示程序。

项目结构

• model.py / dataset.py — 模型与数据集定义
• train.py / train_balanced.py — 标准训练与类别均衡训练
• predict.py — 推理脚本
• visualize.py / visualize_balanced.py — 训练过程可视化
• demo_gui.py — PyQt5 图形界面演示（加载训练好的模型进行预测）
• 实验报告.md — 完整实验报告

运行

pip install torch PyQt5 matplotlib
python train.py            # 训练
python demo_gui.py         # 启动图形界面

说明

• 训练数据（*.csv）与训练好的模型（*.pth、*.pkl）体积较大，未纳入版本库，放入本目录即可运行。
• 训练曲线等结果图见仓库内 png 文件。