python - AttributeError：模块“torchtext.datasets”没有属性“text_classification”

Question

agnews_train, agnews_test = torchtext.datasets.text_classification.DATASETS["AG_NEWS"](root="./datasets")

在上面运行后我得到属性错误。

----> 1 agnews_train，agnews_test = torchtext.datasets.text_classification.DATASETS"AG_NEWS" 中的 AttributeError Traceback（最近一次调用最后一次）

AttributeError：模块“torchtext.datasets”没有属性“text_classification”

Answer 1

您可能只是使用不同的版本。你想要的是：

agnews_train, agnews_test = torchtext.datasets.AG_NEWS(root='./datasets', split=('train', 'test'))

文档