nexuml_library.scenarios.model.linear_ae¶
nexuml_library.scenarios.model.linear_ae
¶
Linear autoencoder model scenario fragments.
linear_ae_reconstruction
¶
linear_ae_reconstruction(
input_dim: int = 128,
hidden_dims: list[int] | None = None,
latent_dim: int = 8,
activation: str = "torch.nn.ReLU",
feature_key: str = "features",
) -> PipelineSpec
Create a PipelineSpec for a linear autoencoder with reconstruction loss.
Returns:
| Name | Type | Description |
|---|---|---|
PipelineSpec |
PipelineSpec
|
Pipeline with linear encoder, decoder and reconstruction loss layers. |
linear_ae_anomaly_detection
¶
linear_ae_anomaly_detection(
input_dim: int = 128,
hidden_dims: list[int] | None = None,
latent_dim: int = 8,
activation: str = "torch.nn.ReLU",
feature_key: str = "features",
score_reduction: str = "mean",
) -> PipelineSpec
Linear autoencoder with reconstruction loss and an anomaly score.
Returns:
| Name | Type | Description |
|---|---|---|
PipelineSpec |
PipelineSpec
|
Pipeline with linear encoder, decoder, reconstruction loss and anomaly score layers. |
linear_ae_lmbe
¶
linear_ae_lmbe(
sample_rate: int = 16000,
n_mels: int = 128,
n_fft: int = 1024,
hop_length: int = 512,
time_frames: int = 128,
hidden_dims: list[int] | None = None,
latent_dim: int = 8,
activation: str = "torch.nn.ReLU",
score_reduction: str = "mean",
) -> PipelineSpec
Waveform -> LMBE -> linear autoencoder with an anomaly score.
Returns:
| Name | Type | Description |
|---|---|---|
PipelineSpec |
PipelineSpec
|
Full waveform-to-reconstruction pipeline with LMBE feature extraction and linear autoencoder. |
linear_ae_multiclass
¶
linear_ae_multiclass(
input_dim: int = 128,
hidden_dims: list[int] | None = None,
latent_dim: int = 8,
activation: str = "torch.nn.ReLU",
) -> PipelineSpec
Create a PipelineSpec for a linear AE with reconstruction + classification.
Returns:
| Name | Type | Description |
|---|---|---|
PipelineSpec |
PipelineSpec
|
Pipeline with encoder, decoder, classification head, reconstruction loss, classification loss and metrics layers. |
linear_ae_multilabel
¶
linear_ae_multilabel(
input_dim: int = 128,
hidden_dims: list[int] | None = None,
latent_dim: int = 8,
activation: str = "torch.nn.ReLU",
) -> PipelineSpec
Create a PipelineSpec for a linear AE with reconstruction + multilabel classification.
Returns:
| Name | Type | Description |
|---|---|---|
PipelineSpec |
PipelineSpec
|
Pipeline with encoder, decoder, multilabel head, reconstruction loss and multilabel loss layers. |
linear_ae_regression
¶
linear_ae_regression(
input_dim: int = 128,
hidden_dims: list[int] | None = None,
latent_dim: int = 8,
num_outputs: int = 3,
activation: str = "torch.nn.ReLU",
) -> PipelineSpec
Create a PipelineSpec for a linear AE with reconstruction + regression.
Returns:
| Name | Type | Description |
|---|---|---|
PipelineSpec |
PipelineSpec
|
Pipeline with encoder, decoder, regression head, reconstruction loss and regression loss layers. |