mltool-0.2.0.0: Machine Learning Toolbox

Copyright(c) Alexander Ignatyev 2017
LicenseBSD-3
Stabilityexperimental
PortabilityPOSIX
Safe HaskellNone
LanguageHaskell2010

MachineLearning.Optimization.MinibatchGradientDescent

Description

Minibatch Gradient Descent

Synopsis

Documentation

minibatchGradientDescent Source #

Arguments

:: Model a 
=> Int

seed

-> Int

batch size

-> R

learning rate, alpha

-> a

model to learn

-> R

epsilon

-> Int

max number of iters

-> Regularization

regularization parameter, lambda

-> Matrix

matrix of features, X

-> Vector

output vector, y

-> Vector

vector of initial weights, theta or w

-> (Vector, Matrix)

vector of weights and learning path

Minibatch Gradient Descent method implementation. See MachineLearning.Regression for usage details.