Renamed model file name

xEcho1337 · Jan 14, 2025 · b3fe3a5 · b3fe3a5
1 parent 1a71b39
commit b3fe3a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test/java/mnist/MNISTClassifier.java b/src/test/java/mnist/MNISTClassifier.java
@@ -28,7 +28,7 @@ public static void main(String[] args) {
     }
 
     public static void evaluateModel(Model model) {
-        model.load("mnist-56.json");
+        model.load("mnist.json");
         model.reloadMatrices();
 
         DataSet set = getData();
@@ -99,7 +99,7 @@ public static void train(Model model, DataSet set) {
             }
 
             if (i % 100 == 0) {
-                model.save("mnist-56.json");
+                model.save("mnist.json");
             }
         }
     }