示例#1
0
        public void SGD(ref MnistData data, int epochs, int batchSize, float learningRate, bool output)
        {
            MathNet.Numerics.Control.UseSingleThread(); // Single thread is optimal over multithreading.

            Console.WriteLine("Training...");

            for (int l = 0; l < vSizes.Count; l++)
            {
                mSumW[l].Clear();
                mSumB[l].Clear();
            }

            RandomizeParameters();

            for (int i = 0; i < epochs; i++)
            {
                for (int j = 0; j < data.TrainImages.ColumnCount; j++)
                {
                    SetInputLayer(data.TrainImages.Column(j));
                    FeedForward();
                    SetOutputLayer((int)(data.TrainLabels[j]));
                    OutputError();
                    Backprop();

                    // Perform summation calculations over the training images.
                    for (int l = lastLayerIndex; l > 0; l--)
                    {
                        mSumW[l] += mError[l].ToColumnMatrix() * mNeurons[l - 1].ToRowMatrix();
                        mSumB[l] += mError[l];
                    }

                    // Update the weights and biases after the mini-batch has been processed.
                    if (j % batchSize == 0 && j > 0)
                    {
                        for (int l = 0; l < vSizes.Count; l++)
                        {
                            mWeights[l] += mSumW[l].Multiply(-learningRate / batchSize);
                            mBiases[l]  += mSumB[l].Multiply(-learningRate / batchSize);
                            mSumW[l].Clear();
                            mSumB[l].Clear();
                        }
                    }
                }
                if (output == true)
                {
                    Console.WriteLine("Epoch {0}: {1} / {2}", i, MnistTest(ref data), data.TestLabels.Count);
                }
            }
            Console.WriteLine("Training Complete.");
        }
示例#2
0
        public int MnistTest(ref MnistData data)
        {
            int correct = 0;
            int total   = data.TestLabels.Count;

            for (int i = 0; i < total; i++)
            {
                SetInputLayer(data.TestImages.Column(i));
                FeedForward();
                if ((mNeurons[lastLayerIndex].MaximumIndex()) == data.TestLabels[i])
                {
                    correct++;
                }
            }
            Console.WriteLine(correct / total);
            this.Accuracy = (double)correct / (double)total;
            return(correct);
        }