nirs4all.operators.models.jax package

Submodules

• nirs4all.operators.models.jax.generic module
  • JaxMLPClassifier
    • JaxMLPClassifier.features
    • JaxMLPClassifier.input_shape
    • JaxMLPClassifier.num_classes
  • JaxMLPRegressor
    • JaxMLPRegressor.features
    • JaxMLPRegressor.input_shape
• nirs4all.operators.models.jax.nicon module
  • Activation
    • Activation.fn
  • DepthwiseConv1D
    • DepthwiseConv1D.activation
    • DepthwiseConv1D.depth_multiplier
    • DepthwiseConv1D.kernel_size
    • DepthwiseConv1D.padding
    • DepthwiseConv1D.strides
  • DynamicModel
    • DynamicModel.layers
  • Flatten
  • GlobalAveragePooling1D
  • MaxPooling1D
    • MaxPooling1D.padding
    • MaxPooling1D.strides
    • MaxPooling1D.window_shape
  • SeparableConv1D
    • SeparableConv1D.activation
    • SeparableConv1D.depth_multiplier
    • SeparableConv1D.features
    • SeparableConv1D.kernel_size
    • SeparableConv1D.padding
    • SeparableConv1D.strides
  • SpatialDropout1D
    • SpatialDropout1D.rate
  • TransformerBlock
    • TransformerBlock.dropout
    • TransformerBlock.ff_dim
    • TransformerBlock.head_size
    • TransformerBlock.num_heads
  • customizable_decon()
  • customizable_decon_classification()
  • customizable_nicon()
  • customizable_nicon_classification()
  • decon()
  • decon_Sep()
  • decon_Sep_classification()
  • decon_classification()
  • decon_layer_classification()
  • get_activation()
  • get_norm()
  • nicon()
  • nicon_VG()
  • nicon_VG_classification()
  • nicon_classification()
  • thin_nicon()
  • transformer()
  • transformer_VG()
  • transformer_VG_classification()
  • transformer_classification()
  • transformer_model()
  • transformer_model_classification()