Source code for gobbli.dataset.trivial
from gobbli.dataset.base import BaseDataset
[docs]class TrivialDataset(BaseDataset):
"""
gobbli Dataset containing only a few observations.
Useful for verifying a model runs without waiting for an
actual dataset to process.
"""
DATASET = ["This is positive.", "This, although, is negative."]
LABELS = ["1", "0"]
def _is_built(self) -> bool:
return True
def _build(self):
pass
[docs] def X_train(self):
return TrivialDataset.DATASET
[docs] def y_train(self):
return TrivialDataset.LABELS
[docs] def X_test(self):
return TrivialDataset.DATASET
[docs] def y_test(self):
return TrivialDataset.LABELS