Package me.yixqiao.jlearn.testing
Class MNIST
- java.lang.Object
-
- me.yixqiao.jlearn.testing.MNIST
-
- Direct Known Subclasses:
MNISTSave
public class MNIST extends java.lang.ObjectTrain a network on the MNIST digits dataset.
-
-
Field Summary
Fields Modifier and Type Field Description protected MatrixevalInputsTest inputs from dataset.protected MatrixevalOutputsTest outputs from dataset.protected MatrixinputsInputs from dataset.protected ModelmodelModel.protected MatrixoutputsOutputs from dataset.
-
Constructor Summary
Constructors Constructor Description MNIST()
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description protected voidbuildModel()Build model.protected voidevaluateModel()Evaluate model.protected voidinitInputs()Initialize inputs.static voidmain(java.lang.String[] args)Run.protected voidprintPredictions()Print predictions.protected voidtrain()Train model.
-
-
-
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.
-
-