+mnist verify on training data

lnashier · Apr 23, 2024 · 808f288 · 808f288
1 parent 3915254
commit 808f288
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/examples/feedforward/mnist/net.go b/examples/feedforward/mnist/net.go
@@ -79,6 +79,13 @@ func Build(ctx context.Context, args []string) {
 		}
 	}
 
+	// verify on training data
+	inputs, targets, err := readData(args[0], args[1])
+	if err != nil {
+		panic(err)
+	}
+	test(ctx, nn, inputs, targets)
+
 	unseenInputs, unseenTargets, err := readData(args[2], args[3])
 	if err != nil {
 		panic(err)