Minor fixes for experiments files

ITMO-NSS-team · Oct 8, 2023 · 5fe4251 · 5fe4251
1 parent eeb0f6a
commit 5fe4251
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 2 deletions.
diff --git a/experiment_burgers.py b/experiment_burgers.py
@@ -135,6 +135,7 @@ def hash_term(term):
         num_found_eq.append(len(difference_ls))
         print('Overall time is:', time1)
         print(f'Iteration processed: {i+1}/{max_iter_number}\n')
+        i += 1
         time_ls.append(time1)
 
     if write_csv:

diff --git a/experiment_burgers_sindy.py b/experiment_burgers_sindy.py
@@ -138,6 +138,7 @@ def hash_term(term):
         num_found_eq.append(len(difference_ls))
         print('Overall time is:', time1)
         print(f'Iteration processed: {i+1}/{max_iter_number}\n')
+        i += 1
         time_ls.append(time1)
 
     if write_csv:

diff --git a/experiment_kdv.py b/experiment_kdv.py
@@ -173,13 +173,13 @@ def hash_term(term):
         if len(difference_ls) != 0:
             differences_ls.append(min(difference_ls))
             mean_diff_ls += difference_ls
-        # else:
-        #     differences_ls.append(None)
+
         num_found_eq.append(len(difference_ls))
 
         print('Overall time is:', time1)
         print(f'Iteration processed: {i+1}/{max_iter_number}\n')
         time_ls.append(time1)
+        i += 1
 
     if write_csv:
         arr = np.array([differences_ls, time_ls, num_found_eq])

diff --git a/experiment_kdv_sindy.py b/experiment_kdv_sindy.py
@@ -140,6 +140,7 @@ def hash_term(term):
         num_found_eq.append(len(difference_ls))
         print('Overall time is:', time1)
         print(f'Iteration processed: {i+1}/{max_iter_number}\n')
+        i += 1
         time_ls.append(time1)
 
     if write_csv:

diff --git a/experiment_wave.py b/experiment_wave.py
@@ -141,6 +141,7 @@ def hash_term(term):
         num_found_eq.append(len(difference_ls))
         print('Overall time is:', time1)
         print(f'Iteration processed: {i+1}/{max_iter_number}\n')
+        i += 1
         time_ls.append(time1)
 
     if write_csv: