最近开始接触机器学习，今天尝试用 Transformers 训练一个文本多分类模型，实现根据标题对新闻进行分类。

文本分类也可用在其他地方，比如垃圾邮件/帖子识别、内容情感分析等，就个人理解，模型的功能和具体表现在于训练数据的内容。

数据读取

这次使用的训练数据集是 今日头条新闻，有 30W+ 条数据，15个分类，训练就想做菜一样，需要提前对食材进行处理，

处理器：AMD 5600G

显卡：Tesla P40

主板：ASUS TUF B450M-PLUS GAMING

PVE 版本：7.2-3

LXC 容器镜像：debian-11-standard_11.6-1_amd64

显卡安装

P40 显卡没有视频输出接口，在 BISO 中检查 Primary Display 选项，使用 IGFX 选项，避免无法显示界面。