| Maintainer | Kiet Lam <ktklam9@gmail.com> |
|---|
AI.Model.General
Description
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]
- initializeGeneral :: [Int] -> Double -> StdGen -> GenericModel
Documentation
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