24 lines
633 B
Python
24 lines
633 B
Python
from .language_modeling import LanguageModelingDataset, WikiText2 # NOQA
|
|
from .snli import SNLI
|
|
from .sst import SST
|
|
from .translation import TranslationDataset, Multi30k, IWSLT, WMT14 # NOQA
|
|
from .sequence_tagging import SequenceTaggingDataset, UDPOS # NOQA
|
|
from .trec import TREC
|
|
from .imdb import IMDb
|
|
from . import generic
|
|
|
|
|
|
__all__ = ['LanguageModelingDataset',
|
|
'SNLI',
|
|
'SST',
|
|
'TranslationDataset',
|
|
'Multi30k',
|
|
'IWSLT',
|
|
'WMT14'
|
|
'WikiText2',
|
|
'TREC',
|
|
'IMDb',
|
|
'SequenceTaggingDataset',
|
|
'UDPOS',
|
|
]
|