Module traits

Source

Modules§

config
predict
train

Traits§

NetworkConfig
Predict
Predict isn’t designed to be implemented directly, rather, as a blanket impl for any entity that implements the Forward trait. This is primarily used to define the base functionality of the Model trait.
PredictWithConfidence
This trait extends the Predict trait to include a confidence score for the prediction. The confidence score is calculated as the inverse of the variance of the output.
Train
This trait defines the training process for the network
TrainingConfiguration