nirs4all.operators.models.jax.nicon module

class nirs4all.operators.models.jax.nicon.Activation(*args: Any, **kwargs: Any)

Bases: Module

fn: Callable

class nirs4all.operators.models.jax.nicon.DepthwiseConv1D(*args: Any, **kwargs: Any)

Bases: Module

activation: Callable | None = None

depth_multiplier: int = 1

kernel_size: int

padding: str = 'SAME'

strides: int = 1

class nirs4all.operators.models.jax.nicon.DynamicModel(*args: Any, **kwargs: Any)

Bases: Module

layers: Sequence[flax.linen.Module]

class nirs4all.operators.models.jax.nicon.Flatten(*args: Any, **kwargs: Any)

Bases: Module

class nirs4all.operators.models.jax.nicon.GlobalAveragePooling1D(*args: Any, **kwargs: Any)

Bases: Module

class nirs4all.operators.models.jax.nicon.MaxPooling1D(*args: Any, **kwargs: Any)

Bases: Module

padding: str = 'SAME'

strides: tuple[int] = (2,)

window_shape: tuple[int] = (2,)

class nirs4all.operators.models.jax.nicon.SeparableConv1D(*args: Any, **kwargs: Any)

Bases: Module

activation: Callable | None = None

depth_multiplier: int = 1

features: int

kernel_size: int

padding: str = 'SAME'

strides: int = 1

class nirs4all.operators.models.jax.nicon.SpatialDropout1D(*args: Any, **kwargs: Any)

Bases: Module

rate: float = 0.5

class nirs4all.operators.models.jax.nicon.TransformerBlock(*args: Any, **kwargs: Any)

Bases: Module

dropout: float

ff_dim: int

head_size: int

num_heads: int

nirs4all.operators.models.jax.nicon.customizable_decon(input_shape, params=None)

nirs4all.operators.models.jax.nicon.customizable_decon_classification(input_shape, num_classes=2, params=None)

nirs4all.operators.models.jax.nicon.customizable_nicon(input_shape, params=None)

nirs4all.operators.models.jax.nicon.customizable_nicon_classification(input_shape, num_classes=2, params=None)

nirs4all.operators.models.jax.nicon.decon(input_shape, params=None)

nirs4all.operators.models.jax.nicon.decon_Sep(input_shape, params=None)

nirs4all.operators.models.jax.nicon.decon_Sep_classification(input_shape, num_classes=2, params=None)

nirs4all.operators.models.jax.nicon.decon_classification(input_shape, num_classes=2, params=None)

nirs4all.operators.models.jax.nicon.decon_layer_classification(input_shape, num_classes=2, params=None)

nirs4all.operators.models.jax.nicon.get_activation(name: str | None)

nirs4all.operators.models.jax.nicon.get_norm(method: str)

nirs4all.operators.models.jax.nicon.nicon(input_shape, params=None)

nirs4all.operators.models.jax.nicon.nicon_VG(input_shape, params=None)

nirs4all.operators.models.jax.nicon.nicon_VG_classification(input_shape, num_classes=2, params=None)

nirs4all.operators.models.jax.nicon.nicon_classification(input_shape, num_classes=2, params=None)

nirs4all.operators.models.jax.nicon.thin_nicon(input_shape, params=None)

nirs4all.operators.models.jax.nicon.transformer(input_shape, params=None)

nirs4all.operators.models.jax.nicon.transformer_VG(input_shape, params=None)

nirs4all.operators.models.jax.nicon.transformer_VG_classification(input_shape, num_classes=2, params=None)

nirs4all.operators.models.jax.nicon.transformer_classification(input_shape, num_classes=2, params=None)

nirs4all.operators.models.jax.nicon.transformer_model(input_shape, params=None)

nirs4all.operators.models.jax.nicon.transformer_model_classification(input_shape, num_classes=2, params=None)