Add forward to layer

opixelum · Mar 11, 2024 · c35fe5a · c35fe5a
1 parent 80beac8
commit c35fe5a
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 15 deletions.
diff --git a/rust/src/ai/layer.rs b/rust/src/ai/layer.rs
@@ -22,17 +22,16 @@ impl Layer {
         }
     }
 
-    // fn forward(&self, inputs: Vec<f64>) -> Vec<f64> {
-    //    self.neurons
-    //        .iter()
-    //        .enumerate()
-    //        .map(|(i, neuron)| {
-    //            if neuron.weights.is_none() {
-    //                self.neurons[i].weights = Some(vec![0.5; inputs.len()]);
-    //            }
+    pub fn forward(&mut self, inputs: Vec<f64>) -> Vec<f64> {
+        let mut outputs = vec![];
 
-    //            neuron.forward(inputs.clone())
-    //        })
-    //        .collect()
-    //}
+        for neuron in self.neurons.iter_mut() {
+            if neuron.weights.is_none() {
+                neuron.weights = Some(vec![1.0; inputs.len()])
+            }
+            outputs.push((self.activation)(neuron.forward(inputs.clone())))
+        }
+
+        outputs
+    }
 }
diff --git a/rust/tests/ai/layer.rs b/rust/tests/ai/layer.rs
@@ -4,15 +4,21 @@ use opixelib::ai::neuron::Neuron;
 
 #[test]
 fn test_new() {
-    let perceptron = Layer::new(1, binary_step);
     assert_eq!(
-        perceptron,
+        Layer::new(1, binary_step),
         Layer {
             neurons: vec![Neuron {
                 weights: None,
                 bias: 0.0
             }],
             activation: binary_step
         }
-    )
+    );
+}
+
+#[test]
+fn test_forward() {
+    let mut perceptron = Layer::new(1, binary_step);
+    assert_eq!(perceptron.forward(vec![1.0, 2.0]), vec![1.0]);
+    assert_eq!(perceptron.forward(vec![-1.0, -2.0]), vec![0.0])
 }