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']¶
-
classmethod
data_dir
()¶ - Return type
Path
-
embed_input
(embed_batch_size=32, pooling=<EmbedPooling.MEAN: 'mean'>, limit=None)¶ - Return type
-
classmethod
load
(*args, **kwargs)¶ - Return type
-
predict_input
(predict_batch_size=32, limit=None)¶ - Return type
-
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
-