MODULE REFERENCE
torch_autoencoder
anomx.models.reconstruction.torch_autoencoderPyTorch Lightning autoencoder model (requires the torch extra).
classView source ↗
TorchAutoencoderModel
TorchAutoencoderModel(config: dict[str, Any] | None = None)Bases: NormalityModel, Reconstructive
Small fully-connected autoencoder trained with PyTorch Lightning.
Reconstruction error per observation lands in model_score. The component exists to establish the torch/lightning integration pattern; the network is intentionally tiny.
| Parameter | Type | Default |
|---|---|---|
config | dict[str, Any] | None | None |
Members
component_keyattribute
component_key = 'torch_autoencoder'component_iconattribute
component_icon = 'CpuChip01'signatureattribute
signature = ModelSignature(structures={DataStructure.TABULAR, DataStructure.TEMPORAL_SEQUENCE}, modalities={Modality.NUMERIC})component_nameattribute
component_name = 'Torch Autoencoder'component_default_configattribute
component_default_config = {'feature_columns': [], 'hidden_size': 8, 'latent_size': 2, 'max_epochs': 5}component_config_schemaattribute
component_config_schema = {'feature_columns': {'type': 'array'}, 'hidden_size': {'type': 'integer'}, 'latent_size': {'type': 'integer'}, 'max_epochs': {'type': 'integer'}}capabilitiesattribute
capabilities: ClassVar[ComponentCapabilities] = ComponentCapabilities(supports_sequences=True)fitmethod
fit(data: Any) -> None| Parameter | Type | Default |
|---|---|---|
data | Any | required |
Returns None
predictmethod
predict(data: Any) -> pd.DataFrame| Parameter | Type | Default |
|---|---|---|
data | Any | required |
Returns pd.DataFrame
savemethod
save(path: str) -> None| Parameter | Type | Default |
|---|---|---|
path | str | required |
Returns None
loadmethod
load(path: str) -> TorchAutoencoderModel| Parameter | Type | Default |
|---|---|---|
path | str | required |
Returns TorchAutoencoderModel
Inherited members
partial_fitNormalityModelscore_batchNormalityModelis_component_abstractBaseComponentget_component_typeBaseComponentget_component_keyBaseComponentget_component_nameBaseComponentget_component_docsBaseComponentget_component_descriptionBaseComponentget_component_iconBaseComponentget_component_image_pathBaseComponentget_component_statusBaseComponentget_component_default_configBaseComponentget_component_config_schemaBaseComponentget_component_code_versionBaseComponentget_component_source_pathBaseComponentget_component_import_pathBaseComponentget_component_capabilitiesBaseComponentget_component_signatureBaseComponentget_component_parametersBaseComponentget_configBaseComponentreconstructReconstructiveget_capability_keyCapability