AttributeError: модуль torchtext.datasets не имеет атрибута text_classification

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

после запуска над строкой я получаю ошибку атрибута.

AttributeError Traceback (последний вызов последним) в ---- ›1 agnews_train, agnews_test = torchtext.datasets.text_classification.DATASETSAG_NEWS

AttributeError: модуль torchtext.datasets не имеет атрибута text_classification


python pytorch nlp torchtext
person DawoodBhai420    schedule 09.06.2021    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Вероятно, вы просто используете другую версию. Вы хотите:

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

Документация

person Mohammed Elmahgiubi    schedule 09.06.2021