gluonts.nursery.few_shot_prediction.src.meta.datasets.registry module#

gluonts.nursery.few_shot_prediction.src.meta.datasets.registry.get_data_module(name: str, **kwargs: Any) → pytorch_lightning.core.datamodule.LightningDataModule[source]#

This method creates the data module with the specified name. The provided keyword arguments must contain ALL arguments required by the model configuration. Superfluous arguments may be provided and are simply ignored.

Parameters

  • name (str) – The canonical name of the data module. See DATA_MODULE_REGISTRY.

  • kwargs (Any) – Keyword arguments passed to the initializer of the data module.

Returns

The data module.

Return type

ModelConfig

gluonts.nursery.few_shot_prediction.src.meta.datasets.registry.register_data_module(cls: gluonts.nursery.few_shot_prediction.src.meta.datasets.registry.M) → gluonts.nursery.few_shot_prediction.src.meta.datasets.registry.M[source]#

Registers the provided class in the global model registry.