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

Question

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

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

<块引用>

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

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

Answer 1

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

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

Documentation