change error function

jefersonjlima · Nov 21, 2023 · cfa9121 · cfa9121
1 parent d885556
commit cfa9121
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/duffing_oscillator_two_unknown_variables_one_state_observed.py b/examples/duffing_oscillator_two_unknown_variables_one_state_observed.py
@@ -33,12 +33,12 @@ def main():
                             'w': 0.9,
                             'beta': 0.1,
                             'w_damping': 0.999,
-                            'escape_min_vel': 0.001,
-                           'escape_min_error': 5e-3},
+                            'escape_min_vel_percent': 0.001,
+                           'escape_min_error': 5e-6},
                 'dyn_system': {'model_path': '',
                                 'external': None,
                                 'state_mask' : [True, False, False],
-                               'loss': 'mse',
+                               'loss': 'rmse',
                                 'x0': [0., 0., 0.],
                                 't': [0,50,500]
                                 }