AI.Model.Classification

This module provides an initialization for a classification neural network model

NOTE: This theoretically should be faster than the General model if used for classification

Documentation

initializeClassification

Arguments

:: [Int]

The architecture of the neural network

-> Double

The regularization constant

-> StdGen

The random generator

-> GenericModel

Returns the initialized classification model

Make a neural network model that should be used for classification using the Sigmoid as the activation model and Logistic as the cost model