Merge pull request #158 from MatisPatel/fixing--LV

Fixing  lv equations in dynamics.ode and psl.autonomous

src/neuromancer/dynamics/ode.py

@@ -287,8 +287,8 @@ def __init__(self, insize=2, outsize=2):
     def ode_equations(self, x):
         x1 = x[:, [0]]
         x2 = x[:, [-1]]
-        dx1 = x1 - 0.1*x1*x2
-        dx2 = 1.5*x2 + 0.75*0.1*x1*x2
+        dx1 = self.alpha*x1 - 0.4*x1*x2
+        dx2 = 0.1*x1*x2 - 0.4*x2
         return torch.cat([dx1, dx2], dim=-1)
 
 
@@ -423,4 +423,4 @@ def ode_equations(self, x, u):
 odes = {**ode_param_systems_auto,
         **ode_param_systems_nonauto,
         **ode_hybrid_systems_auto,
-        **ode_networked_systems}
+        **ode_networked_systems}

src/neuromancer/psl/autonomous.py

@@ -190,17 +190,17 @@ class LotkaVolterra(ODE_Autonomous):
     def params(self):
         variables = {'x0': [5., 100.]}
         constants = {'ts': 0.1}
-        parameters = {'a': 1.,
-                      'b': 0.1,
-                      'c': 1.5,
-                      'd': 0.75,}
+        parameters = {'a': 1.1,
+                      'b': 0.4,
+                      'c': 0.1,
+                      'd': 0.4,}
         meta = {}
         return variables, constants, parameters, meta
 
     @cast_backend
     def equations(self, t, x):
         dx1 = self.a*x[0] - self.b*x[0]*x[1]
-        dx2 = -self.c*x[1] + self.d*self.b*x[0]*x[1]
+        dx2 = self.c**x[0]*x[1] - self.d*x[1]
         return [dx1, dx2]