nirs4all.operators.models.jax.generic module

class nirs4all.operators.models.jax.generic.JaxMLPClassifier(*args: Any, **kwargs: Any)[source]

Bases: Module

Simple MLP Classifier using Flax.

features: Sequence[int]
input_shape: Any | None = None
num_classes: int
class nirs4all.operators.models.jax.generic.JaxMLPRegressor(*args: Any, **kwargs: Any)[source]

Bases: Module

Simple MLP Regressor using Flax.

features: Sequence[int]
input_shape: Any | None = None