dymad.models.helpers

Functions

build_autoencoder(model_config, dims, dtype, ...)

build_model(model_spec, model_config, data_meta)

Build a model from a typed ModelSpec.

build_model_from_spec(model_spec, ...[, ...])

Compatibility alias for the typed-only build path.

build_processor(model_config, dims, dtype, ...)

fzu_selector(fzu_type, ...)

get_dims(model_config, data_meta)

Determine dimensions used in the model based on configuration and metadata.

dymad.models.helpers.build_autoencoder(model_config, dims, dtype, device, ifgnn=False)
dymad.models.helpers.build_model(model_spec, model_config, data_meta, dtype=None, device=None)

Build a model from a typed ModelSpec.

dymad.models.helpers.build_model_from_spec(model_spec, model_config, data_meta, dtype=None, device=None)

Compatibility alias for the typed-only build path.

dymad.models.helpers.build_processor(model_config, dims, dtype, device, ifgnn=False)
dymad.models.helpers.fzu_selector(fzu_type, n_total_control_features, const_term)
dymad.models.helpers.get_dims(model_config, data_meta)

Determine dimensions used in the model based on configuration and metadata.

This is a generic guess and can be overridden by specific model classes.