SteepestDescentWithMomentumTraining.cs

Go to the documentation of this file.

namespace SpikingNeuronNetwork.Lib.Training
{
    using Interfaces;

    public class SteepestDescentWithMomentumTraining : ITrainingAlgorithm
    {
        public void UpdateWeights(TrainingStats trainingStats, IErrorDerivativeParameters errorDerivativeParameters,
                                  IErrorDerivativeParameters previousErrorDerivativeParameters)
        {
            foreach (var synapse in errorDerivativeParameters.GetSynapses())
            {
                trainingStats.CurrentNetwork.SetSynapticWeight(synapse,
                                        trainingStats.CurrentNetwork.GetSynapticWeight(synapse) -
                                        trainingStats.LearningRate * errorDerivativeParameters.GetErrorDerivative(synapse) -
                                        trainingStats.MomentumLearningRate * (previousErrorDerivativeParameters == null ? 0 : previousErrorDerivativeParameters.GetErrorDerivative(synapse)));
            }
        }
    }
}