flambe.nlp.classification.datasets
¶
Module Contents¶
-
class
flambe.nlp.classification.datasets.
SSTDataset
(binary: bool = True, phrases: bool = False, cache: bool = True, transform: Dict[str, Union[Field, Dict]] = None)[source]¶ Bases:
flambe.dataset.TabularDataset
The official SST-1 dataset.
-
class
flambe.nlp.classification.datasets.
TRECDataset
(cache: bool = True, transform: Dict[str, Union[Field, Dict]] = None)[source]¶ Bases:
flambe.dataset.TabularDataset
The official TREC dataset.
-
class
flambe.nlp.classification.datasets.
NewsGroupDataset
(cache: bool = False, transform: Dict[str, Union[Field, Dict]] = None)[source]¶ Bases:
flambe.dataset.TabularDataset
The official 20 news group dataset.