Class MNIST

  • Direct Known Subclasses:
    MNISTSave

    public class MNIST
    extends java.lang.Object
    Train a network on the MNIST digits dataset.
    • Field Detail

      • inputs

        protected Matrix inputs
        Inputs from dataset.
      • outputs

        protected Matrix outputs
        Outputs from dataset.
      • evalInputs

        protected Matrix evalInputs
        Test inputs from dataset.
      • evalOutputs

        protected Matrix evalOutputs
        Test outputs from dataset.
      • model

        protected Model model
        Model.
    • Constructor Detail

      • MNIST

        public MNIST()
    • Method Detail

      • main

        public static void main​(java.lang.String[] args)
        Run.
        Parameters:
        args - args
      • initInputs

        protected void initInputs()
        Initialize inputs.
      • buildModel

        protected void buildModel()
        Build model.
      • train

        protected void train()
        Train model.
      • evaluateModel

        protected void evaluateModel()
        Evaluate model.
      • printPredictions

        protected void printPredictions()
        Print predictions.