sim-ballena 0.1.2__tar.gz → 0.1.3__tar.gz

{sim_ballena-0.1.2 → sim_ballena-0.1.3}/Cargo.lock

@@ -4,7 +4,7 @@ version = 4
 
 [[package]]
 name = "SimBallenaSNN"
-version = "0.1.2"
+version = "0.1.3"
 dependencies = [
  "pyo3",
  "rand",

{sim_ballena-0.1.2 → sim_ballena-0.1.3}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "SimBallenaSNN"
-version = "0.1.2"
+version = "0.1.3"
 edition = "2024"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

{sim_ballena-0.1.2 → sim_ballena-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sim_ballena
-Version: 0.1.2
+Version: 0.1.3
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

{sim_ballena-0.1.2 → sim_ballena-0.1.3}/src/responses.rs

@@ -8,6 +8,7 @@ const MAX_PRECISION     :f64 = 100_000.0;
 /* ======= OBJECTS  ========= */
 /* ========================== */
 #[derive(Clone)]
+#[derive(Debug)]
 struct VoltageMarker{
     pub t : f64,
     pub v : f64,
@@ -198,13 +199,13 @@ impl Response{
         let tau_list          = self.tau_list.as_ref().unwrap();
         let v_rest_list       = self.v_rest_list.as_ref().unwrap();
 
-        for i in 0..self.outputs.len(){
+        for (o_idx,o) in self.outputs.iter().enumerate(){ 
             let markers_serie:Vec<&VoltageMarker> = self.voltages.iter()
-                                                                .filter(|vm|vm.id==i)
+                                                                .filter(|vm|vm.id==*o)
                                                                 .collect();
 
-            let tau           = *tau_list.get(i).unwrap();
-            let v_rest        = *v_rest_list.get(i).unwrap();
+            let tau           = *tau_list.get(o_idx).unwrap();
+            let v_rest        = *v_rest_list.get(o_idx).unwrap();
             voltages.push( self.extract_voltage_serie(markers_serie, simulation_times, tau, v_rest) );
         }
 

sim_ballena-0.1.3/test_file.py

@@ -0,0 +1,37 @@
+import sim_ballena as ballena
+import matplotlib.pyplot as plt
+
+
+net = (ballena.Network( [ballena.Lif().tau(5),ballena.Lif().tau(0.2)] )
+       .synapses_net([])
+       .weights_net([])
+       .outputs([1])
+       .mode(['VOLTAGES','SPIKES']))
+
+
+
+# ==========
+#   STEP 1
+# ==========
+net = net.synapses_in([(0,0),(0,1)])
+net = net.weights_in([10,10])
+
+times = [ (1, 0) ]
+instance = ballena.Instance( times )
+res = net.simulate( instance, 2 )
+
+time = res.time()
+volt = res.voltages()
+
+plt.figure(figsize=(10,5))
+plt.subplot(121)
+plt.ylim(-80,-55)
+plt.plot( time, volt[0], label='true')
+# plt.subplot(122)
+# plt.ylim(-80,-55)
+# plt.plot( time, volt[1], label='true')
+
+plt.show()
+
+
+

sim_ballena-0.1.2/test_file.py

@@ -1,49 +0,0 @@
-import sim_ballena as ballena
-import matplotlib.pyplot as plt
-
-
-net = (ballena.Network( ballena.Lif().tau(5).t_refractory(0).repeat(2) )
-       .synapses_net([])
-       .weights_net([])
-       .outputs([0,1])
-       .mode(['VOLTAGES','SPIKES']))
-
-
-
-# ==========
-#   STEP 1
-# ==========
-net = net.synapses_in([(0,0),(0,1),(1,0),(1,1)])
-net = net.weights_in([16,0,-16,0])
-
-times = [ (1, 0), (5, 1) ]
-instance = ballena.Instance( times )
-res = net.simulate( instance, 10 )
-
-time = res.time()
-volt = res.voltages() 
-plt.figure(figsize=(10,4))
-plt.subplot(121)
-plt.plot( time, volt[0], label='true')
-plt.subplot(122)
-plt.plot( time, volt[1], label='false')
-plt.show()
-
-# ==========
-#   STEP 1
-# ==========
-net = net.weights_in([16,0,-16,0])
-
-times = [ (1, 0), (5, 1) ]
-instance = ballena.Instance( times )
-res = net.simulate( instance, 10 )
-
-time = res.time()
-volt = res.voltages() 
-plt.figure(figsize=(10,4))
-plt.subplot(121)
-plt.plot( time, volt[0], label='true')
-plt.subplot(122)
-plt.plot( time, volt[1], label='false')
-plt.show()
-