Model Factory
Model creation factory functions.
model_factory(model_configuration)
Deep learning model factory.
Supported models are defined in careamics.config.SupportedArchitecture.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
model_configuration
|
Union[UNetModel, VAEModel]
|
Model configuration. |
required |
Returns:
| Type | Description |
|---|---|
Module
|
Model class. |
Raises:
| Type | Description |
|---|---|
NotImplementedError
|
If the requested architecture is not implemented. |