ErrorDerivativeParametersBatch.cs

Go to the documentation of this file.

namespace SpikingNeuronNetwork.Lib.Training
{
    using Interfaces;
    using System;
    using System.Collections.Generic;
    using System.Linq;

    public class ErrorDerivativeParametersBatch : IErrorDerivativeParameters
    {
        public Dictionary<Synapse, Tuple<double, int>> ErrorDerivativeTotal { get; set; }

        public ErrorDerivativeParametersBatch(ErrorDerivativeParameters errorDerivativeParameters)
        {
            ErrorDerivativeTotal = new Dictionary<Synapse, Tuple<double, int>>();
            foreach (var synapse in errorDerivativeParameters.GetSynapses())
            {
                ErrorDerivativeTotal.Add(synapse, new Tuple<double, int>(errorDerivativeParameters.GetErrorDerivative(synapse), 1));
            }
        }

        public void AddErrorDerivative(Synapse synapse, double errorDerivativeValue)
        {
            if (ErrorDerivativeTotal.ContainsKey(synapse))
            {
                ErrorDerivativeTotal[synapse] = new Tuple<double, int>(
                    ErrorDerivativeTotal[synapse].Item1 + errorDerivativeValue,
                    ErrorDerivativeTotal[synapse].Item2 + 1);
            }
            else
            {
                ErrorDerivativeTotal.Add(synapse, new Tuple<double, int>(errorDerivativeValue, 1));
            }            
        }

        public List<Synapse> GetSynapses()
        {
            return ErrorDerivativeTotal.Keys.ToList();
        }

        public double GetErrorDerivative(Synapse synapse)
        {
            return ErrorDerivativeTotal[synapse].Item1 / ErrorDerivativeTotal[synapse].Item2;
        }
    }
}