dataset.rst

Dataset API

All datset implementations inherit from the the abstract class :py~small_text.data.datasets.Dataset. Several such implementations are available, depending on the choice of classifier (and on the installed optional dependencies <install:Optional Dependencies>).

Overview

---

Core

small_text.data.datasets

Dataset

x

y

target_labels

small_text.data.datasets.SklearnDataset

small_text.data.datasets.TextDataset

---

Pytorch Integration

small_text.integrations.pytorch.datasets

PytorchTextClassificationDataset

---

Transformers Integration

small_text.integrations.transformers.datasets

TransformersDataset