TrainingStats.cs

Go to the documentation of this file.

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

    public class TrainingStats
    {
        public int Id { get; set; }

        public SpikingNeuronNetwork OriginalNetwork { get; set; }

        public TrainingMethod TrainingMethod { get; set; }

        public ITrainingAlgorithm TrainingAlgorithm { get; set; }

        public double LearningRate { get; set; }

        public double MomentumLearningRate { get; set; }

        public double MaximumErrorAfterTraining { get; set; }

        public double MaximumNumberOfTrainingEpochs { get; set; }

        public List<SpikeSet> TrainingSet { get; set; }

        public ISpikingError ErrorCalculator { get; set; }

        public int NumEpochs { get; set; }

        public Dictionary<int, TrainingStatsPerEpoch> PerEpochStats { get; set; }

        public SpikingNeuronNetwork CurrentNetwork { get; set; }

        public void ResetTraining()
        {
            CurrentNetwork = OriginalNetwork.Clone();
            NumEpochs = 0;
            PerEpochStats = new Dictionary<int, TrainingStatsPerEpoch>();            
        }
    }
}