fixed a bug in the optimizer. Continue to work on the notebook

dynamicslab · Mar 27, 2023 · d02f4c2 · d02f4c2
1 parent 330a28d
commit d02f4c2
Show file tree

Hide file tree

Showing 2 changed files with 181 additions and 908 deletions.
diff --git a/examples/17_trapping_inequality.ipynb b/examples/17_trapping_inequality.ipynb
diff --git a/pysindy/optimizers/trapping_sr3.py b/pysindy/optimizers/trapping_sr3.py
@@ -376,8 +376,7 @@ def _set_Ptensors(self, r):
         PC_tensor = np.zeros((r, r, N))
         if offset:
             for i in range(r):
-                for j in range(r):
-                    PC_tensor[i, j, 0] = 1
+                PC_tensor[i, i, 0] = 1.0
 
         # delta_{il}delta_{jk}
         PL_tensor = np.zeros((r, r, r, N))