AI.Model.General

This module provides an initialization for a general neural network model that can do either regression or classification

If for regression, the training data must be normalized by user to have range of [-1,1]

Documentation

initializeGeneral

Arguments

:: [Int]

The architecture of the neural network

-> Double

The regularization constant

-> StdGen

The random generator

-> GenericModel

Returns the initialized general model

This is a general neural network model that can be used for classification or regression using HyperbolicTangent as the activation model and MeanSquared as the cost model