from torchtext.datasets import text_classification 报错解决：cannot import name ‘text_classification‘(代码片

Posted 2021-10-22 ZSYL

from torchtext.datasets import text_classification 报错解决

• 问题描述
• 解决方法

问题描述

做文本分类的时候发现这个问题：from torchtext.datasets import text_classification

报错:

ImportError: cannot import name 'text_classification' from 'torchtext.datasets'

阅读源码发现没有这个模块，查看官方文档发现新版本较大改动。

解决方法

更改导入方式：

torchtext.datasets.AG_NEWS(root='.data', split=('train', 'test'))

# 注释掉from torchtext.datasets import text_classification 
# 原代码改为：
train_dataset, test_dataset = torchtext.datasets.AG_NEWS(root='./data/ag_news_csv/', split=('train', 'test'))

# 原代码：
# train_dataset, test_dataset = text_classification.DATASETS['AG_NEWS'](root=load_data_path)

即可解决！

参考：Link

加油!

感谢!

努力!