Module nn 

This module provides network specific implementations and traits supporting the development of neural network models.

Modules

layer

Structs

LayerBase

The LayerBase implementation works to provide a generic interface for layers within a neural network by associating an activation function F with a set of parameters P.

Traits

Model

The Model trait defines the core interface for all models; implementors will need to provide the type of configuration used by the model, the type of layout used by the model, and the type of parameters used by the model. The crate provides standard, or default, definitions of both the configuration and layout types, however, for

ModelExt

NetworkConfig

The NetworkConfig trait defines an interface for compatible configurations within the framework, providing a layout and a key-value store to manage hyperparameters.

NetworkConsts

A trait defining common constants for neural networks.

NetworkParams

NeuralNetwork

The NeuralNetwork trait is used to define the network itself as well as each of its constituent parts.

RawContext

RawLayer

The RawLayer trait establishes a common interface for all layers within a given model. Implementors will need to define the type of parameters they utilize, as well as provide methods to access both the activation function and the parameters of the layer.

RawLayerMut

The RawLayerMut trait extends the RawLayer trait by providing mutable access to the layer’s parameters and additional methods for training the layer, such as backward propagation and parameter updates.

Type Aliases

FnLayer

A dynamic, functional alias of the [Layer] implementation leveraging boxed closures.

HeavySideLayer

A [Layer] type using the heavyside activation function.

LayerDyn

A dynamic instance of the layer using a boxed activator.

LayerParams

A type alias for an owned [Layer] configured to use the standard Params instance

LayerParamsBase

A type alias for a layer configured to use the ParamsBase instance

LinearLayer

A type alias for a layer using a linear activation function.

ReluLayer

A [Layer] type using the ReLU activation function.

SigmoidLayer

A type alias for a [Layer] using a sigmoid activation function.

TanhLayer

An alias for a [Layer] that uses the hyperbolic tangent function.