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