Module model

This module provides the scaffolding for creating models and layers in a neural network.

Modules

config

store

Structs

ModelParams

This object is an abstraction over the parameters of a deep neural network model. This is done to isolate the necessary parameters from the specific logic within a model allowing us to easily create additional stores for tracking velocities, gradients, and other metrics we may need.

StandardModelConfig

Traits

Model

This trait defines the base interface for all models, providing access to the models configuration, layout, and learned parameters.