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.

URL = https://raw.githubusercontent.com/harvardnlp/sent-conv-torch/master/data/[source]
classmethod _load_file(cls, path: str, sep: Optional[str] = 't', header: Optional[str] = None, columns: Optional[Union[List[str], List[int]]] = None, encoding: Optional[str] = 'utf-8')[source]

Load data from the given path.

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.

URL = https://raw.githubusercontent.com/harvardnlp/sent-conv-torch/master/data/[source]
classmethod _load_file(cls, path: str, sep: Optional[str] = 't', header: Optional[str] = None, columns: Optional[Union[List[str], List[int]]] = None, encoding: Optional[str] = 'latin-1')[source]

Load data from the given path.

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.