Update README.md

unda-ml · Apr 16, 2024 · 434a6d4 · 434a6d4
1 parent 57421dc
commit 434a6d4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -47,18 +47,18 @@ fn main() {
 
     let mut new_net = Network::new(4);
 
-    new_net.set_input(InputTypes::DENSE(2))
+    new_net.set_input(InputTypes::DENSE(2));
     new_net.add_layer(LayerTypes::DENSE(3, Activations::RELU, 0.001));
     new_net.add_layer(LayerTypes::DENSE(1, Activations::SIGMOID, 0.001));
 
     new_net.compile();
 
     new_net.fit(&inputs, &outputs, 2, ErrorTypes::MeanAbsolute);
 
-    println!("1 and 0: {:?}", new_net.predict(vec![1.0,0.0])[0]);
-    println!("0 and 1: {:?}", new_net.predict(vec![0.0,1.0])[0]);
-    println!("1 and 1: {:?}", new_net.predict(vec![1.0,1.0])[0]);
-    println!("0 and 0: {:?}", new_net.predict(vec![0.0,0.0])[0]);
+    println!("1 and 0: {:?}", new_net.predict(&vec![1.0,0.0])[0]);
+    println!("0 and 1: {:?}", new_net.predict(&vec![0.0,1.0])[0]);
+    println!("1 and 1: {:?}", new_net.predict(&vec![1.0,1.0])[0]);
+    println!("0 and 0: {:?}", new_net.predict(&vec![0.0,0.0])[0]);
 
     new_net.save("best_network.json");
 }