Package me.yixqiao.jlearn.testing
Class MNISTSave
- java.lang.Object
-
- me.yixqiao.jlearn.testing.MNIST
-
- me.yixqiao.jlearn.testing.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.
-
-
Field Summary
-
Fields inherited from class me.yixqiao.jlearn.testing.MNIST
evalInputs, evalOutputs, inputs, model, outputs
-
-
Constructor Summary
Constructors Constructor Description MNISTSave()
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description protected void
loadModel()
Load the model.static void
main(java.lang.String[] args)
Run.protected static void
runLoad()
Load and evaluate a model.protected static void
runTrain()
Train a new model.protected void
train()
Train model.-
Methods inherited from class me.yixqiao.jlearn.testing.MNIST
buildModel, evaluateModel, initInputs, printPredictions
-
-