Update initial_NM.py

inamoto85 · Sep 7, 2022 · 6187a01 · 6187a01
1 parent e71c82f
commit 6187a01
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nm_planner/initial_NM.py b/nm_planner/initial_NM.py
@@ -1,7 +1,7 @@
 import pandas as pd
 
 
-def init_data(patient_id: str, n_init: int, mode="file", file="pars.csv"):
+def init_data(patient_id: str, n_init: int, mode="file", file="/util_sobol/pars.csv"):
     """data initialization for the BO procedure
     Args:
         patient_id: str,
@@ -22,4 +22,4 @@ def init_data(patient_id: str, n_init: int, mode="file", file="pars.csv"):
             value = sobol_odf.iloc[i, -1]
             pars.append(par)
             vals.append(value)
-    return pars, vals
+    return pars, vals