Add more inputs to test

unda-ml · Feb 12, 2024 · 3a486d4 · 3a486d4
1 parent 94ea0c0
commit 3a486d4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -16,23 +16,23 @@ async fn main() {
     println!("Done Generating MNIST");
 
     outputs = to_categorical(outputs_uncat);
-    for i in 0..600{
+    for i in 0..3000{
         inputs.push(&inputs_undyn[i]);
         true_outputs.push(outputs[i].clone());
     }
     loop{
         let mut network = Network::new(128);
 
         network.set_input(InputTypes::DENSE(784));
+        network.add_layer(LayerTypes::DENSE(256, Activations::RELU, 0.001));
         network.add_layer(LayerTypes::DENSE(64, Activations::RELU, 0.001));
-        network.add_layer(LayerTypes::DENSE(32, Activations::RELU, 0.001));
         network.add_layer(LayerTypes::DENSE(10, Activations::SOFTMAX, 0.001));
 
         //network.set_log(false);
 
         network.compile();
 
-        network.fit(&inputs, &true_outputs, 5, ErrorTypes::CategoricalCrossEntropy);
+        network.fit(&inputs, &true_outputs, 1, ErrorTypes::CategoricalCrossEntropy);
         for i in 0..10{
             println!("predicted: {:?} \n\n actual: {:?}\n", network.predict(inputs[i]), true_outputs[i]);
         }