torchtext.datasets — torchtext 0.11.0 documentation
pytorch.org › text › stabletorchtext.datasets.AG_NEWS (root='.data', split=('train', 'test')) [source] ¶ AG_NEWS dataset. Separately returns the train/test split. Number of lines per split: train: 120000. test: 7600. Number of classes. 4. Parameters. root – Directory where the datasets are saved. Default: .data. split – split or splits to be returned. Can be a ...