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.Object
Train a network on the MNIST digits dataset.
-
-
Field Summary
Fields Modifier and Type Field Description protected Matrix
evalInputs
Test inputs from dataset.protected Matrix
evalOutputs
Test outputs from dataset.protected Matrix
inputs
Inputs from dataset.protected Model
model
Model.protected Matrix
outputs
Outputs from dataset.
-
Constructor Summary
Constructors Constructor Description MNIST()
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description protected void
buildModel()
Build model.protected void
evaluateModel()
Evaluate model.protected void
initInputs()
Initialize inputs.static void
main(java.lang.String[] args)
Run.protected void
printPredictions()
Print predictions.protected void
train()
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.
-
-