admin管理员组

文章数量:1593939

AG_NEWS 新闻主题分类任务(PyTorch版)

  • 前言
  • 1. 使用 N 元组加载数据
  • 2. 安装 Torch-GPU&TorchText
  • 3. 访问原始数据集迭代器
  • 4. 准备数据处理管道
  • 5. 生成数据批次和迭代器
  • 6. 定义模型
  • 7. 初始化一个实例
  • 8. 定义训练模型和评估结果的函数
  • 9. 分割数据集并运行模型
  • 10. 测试随机新闻
  • 11. 完整代码
  • 总结

前言

这是TorchText官方的一个教程,更多内容.请参考官方文档:PyTorch / TorchText .

本教程说明如何使用torchtext中的文本分类数据集,包括:

- AG_NEWS,
- SogouNews,
- DBpedia,
- YelpReviewPolarity,
- YelpR

本文标签: 实战文本主题新闻NLP