CondVAEModel.from_datamodule

classmethod CondVAEModel.from_datamodule(datamodule: DataModule, **kwargs) → CondAEModel[source]

Create a model from a data module.

Parameters:

  • datamodule (DataModule) – The data module object that provides the training, validation, and test data.

  • **kwargs – Additional keyword arguments that can be passed to the model. See CondAEModel.__init__() for more details.

Returns:

CondAEModel – The model object.