天池新闻文本分类pytorch
时间: 2023-10-31 17:07:17 浏览: 130
天池新闻文本分类pytorch是一个用于实现新闻文本分类的pytorch框架。在这个框架中,作者采用了"预训练 fgm 交叉验证模型融合"的解决方案。具体来说,作者使用了bert4keras作为基础框架,通过预训练模型和快速梯度方法(fgm)对模型进行训练和优化。这个解决方案在天池的新闻文本分类比赛中获得了Top1的成绩。此外,作者还根据bert4torch复现了"fgm 交叉验证模型融合"的部分,并在不同参数下进行了多次复现实验,取得了不错的结果。如果你对这个解决方案感兴趣,可以参考原作者的项目链接以获取更详细的信息。
阅读全文