gt4sd.frameworks.granular.ml.models.model_builder module

Model builder module.

Summary

Functions:

build_model	Build model from architecture configuration.
building_models	Building models given architecture configurations.
define_latent_models_input_size	Define latent models input size and return the updated configurations.

Reference

build_model(architecture)

Build model from architecture configuration.

Parameters

architecture (Dict[str, Any]) – architecture configuration.

Return type

GranularBaseModel

Returns

built model.

building_models(architectures)

Building models given architecture configurations.

Parameters

architectures (List[Dict[str, Any]]) – list of architecture configurations.

Return type

List[GranularBaseModel]

Returns

a list of models.

define_latent_models_input_size(architecture_autoencoders, architecture_latent_models)

Define latent models input size and return the updated configurations.

Parameters

• architecture_autoencoders (List[Dict[str, Any]]) – list of autoencoder architecture configurations.

• architecture_latent_models (List[Dict[str, Any]]) – list of latent model architecture configurations.

Return type

List[Dict[str, Any]]

Returns

list of update latent model architecture configurations.