fastar.nn.pca_regressor

Classes

PCARegressor(output_dim, activation_type, ...)

Simple feed-forward neural network for predicting PCA coefficients from stellar parameters.

class fastar.nn.pca_regressor.PCARegressor(output_dim: int = 16, activation_type: str = 'gelu', parent: Module | Scope | _Sentinel | None = <flax.linen.module._Sentinel object>, name: str | None = None)

Bases: Module

Simple feed-forward neural network for predicting PCA coefficients from stellar parameters.

Notes

Todo

Fix the warning related with the Sphinx when the attributes are added to the class PCARegressor.

activation_type: str = 'gelu'

name: str | None = None

output_dim: int = 16

parent: Module | Scope | _Sentinel | None = None

scope: Scope | None = None