gobbli.dataset.trivial module

class gobbli.dataset.trivial.TrivialDataset(*args, **kwargs)[source]

Bases: gobbli.dataset.base.BaseDataset

gobbli Dataset containing only a few observations. Useful for verifying a model runs without waiting for an actual dataset to process.

Blank constructor needed to satisfy mypy

DATASET = ['This is positive.', 'This, although, is negative.']
LABELS = ['1', '0']
X_test()[source]
X_train()[source]
classmethod data_dir()
Return type

Path

embed_input(embed_batch_size=32, pooling=<EmbedPooling.MEAN: 'mean'>, limit=None)
Return type

EmbedInput

classmethod load(*args, **kwargs)
Return type

BaseDataset

predict_input(predict_batch_size=32, limit=None)
Return type

PredictInput

train_input(train_batch_size=32, valid_batch_size=8, num_train_epochs=3, valid_proportion=0.2, split_seed=1234, shuffle_seed=1234, limit=None)
Return type

TrainInput

y_test()[source]
y_train()[source]