Class MNISTSave


  • public class MNISTSave
    extends MNIST
    Train a model on the MNIST dataset, then save it to a file and load it back to evaluate.
    • Constructor Detail

      • MNISTSave

        public MNISTSave()
    • Method Detail

      • main

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

        protected static void runTrain()
        Train a new model.
      • runLoad

        protected static void runLoad()
        Load and evaluate a model.
      • loadModel

        protected void loadModel()
        Load the model.
      • train

        protected void train()
        Description copied from class: MNIST
        Train model.
        Overrides:
        train in class MNIST