Create net struct

Author: sumedhe

src/core/mod.rs

@@ -1,4 +1,5 @@
 mod som;
-mod functions;
+mod net;
 
-pub use self::som::SOM;
+pub use self::som::SOM;
+pub use self::net::Net;

src/core/net.rs

@@ -0,0 +1,15 @@
+// Neural network struct
+pub struct Net {
+  pub nodes: Vec<Vec<i32>>,
+  pub size: (u32, u32)
+}
+
+impl Net {
+  // Create new network
+  pub fn new(size: (u32, u32), dimension: u32) -> Net {
+    Net {
+      nodes: vec![vec![0i32; dimension as usize]; (size.0 * size.1) as usize],
+      size: size
+    }
+  }
+}

src/core/som.rs

@@ -1,4 +1,5 @@
 use crate::enums::*;
+use crate::core::*;
 
 // SOM Struct
 pub struct SOM {
@@ -25,8 +26,29 @@ impl SOM {
     }
   }
 
-  // Start train
+  // Start training
   pub fn train(&self, dataset: &Vec<&Vec<i32>>, iterations: i32) -> () {
+    let dimension: u32 = 3; // Dimension
+    let _net: Net = Net::new(self.size, dimension);
+
+    // For each iteration
+    for _i in 0..iterations {
+      // Foe each dataitem
+      for _item in dataset {
+
+        // println!("{}", item[0])
+      }
+
+      // println!("{}", i);
+    }
+
+
+
+
+
+
     println!("Traing {} {}", dataset[1][2], iterations);
+
+
   }
 }

src/lib.rs

@@ -1,5 +1,7 @@
-mod enums;
 mod core;
+mod enums;
+mod helpers;
 
-pub use crate::enums::*;
 pub use crate::core::*;
+pub use crate::enums::*;
+pub use crate::helpers::*;