gobbli.dataset.base module¶
-
class
gobbli.dataset.base.
BaseDataset
(*args, **kwargs)[source]¶ Bases:
abc.ABC
Abstract base class for datasets used for benchmarking and testing.
- Derived classes should account for the following:
Dataset order should be consistent so limiting can work correctly
Blank constructor needed to satisfy mypy
-
embed_input
(embed_batch_size=32, pooling=<EmbedPooling.MEAN: 'mean'>, limit=None)[source]¶ - Return type
-
gobbli.dataset.base.
LOGGER
= <Logger gobbli.dataset.base (WARNING)>¶