Merge pull request #80 from jxx123/fix_warning

Fix random init state
jxx123 · May 16, 2024 · a7e68d6 · a7e68d6
2 parents 61919d7 + 25b1bab
commit a7e68d6
Show file tree

Hide file tree

Showing 85 changed files with 14,941 additions and 14,627 deletions.
diff --git a/.gitignore b/.gitignore
@@ -2,7 +2,7 @@
 *.pyc
 
 # results foler
-/results/
+/examples/results
 
 # Setuptools distribution folder.
 /dist/

diff --git a/examples/results/2017-12-31_17-46-32/BG_trace.png b/examples/results/2017-12-31_17-46-32/BG_trace.png
diff --git a/examples/results/2017-12-31_17-46-32/CVGA.png b/examples/results/2017-12-31_17-46-32/CVGA.png
diff --git a/examples/results/2017-12-31_17-46-32/CVGA_stats.csv b/examples/results/2017-12-31_17-46-32/CVGA_stats.csv
diff --git a/examples/results/2017-12-31_17-46-32/adolescent#001.csv b/examples/results/2017-12-31_17-46-32/adolescent#001.csv
diff --git a/examples/results/2017-12-31_17-46-32/adolescent#002.csv b/examples/results/2017-12-31_17-46-32/adolescent#002.csv
diff --git a/examples/results/2017-12-31_17-46-32/adolescent#003.csv b/examples/results/2017-12-31_17-46-32/adolescent#003.csv
diff --git a/examples/results/2017-12-31_17-46-32/adolescent#004.csv b/examples/results/2017-12-31_17-46-32/adolescent#004.csv
diff --git a/examples/results/2017-12-31_17-46-32/adolescent#005.csv b/examples/results/2017-12-31_17-46-32/adolescent#005.csv
diff --git a/examples/results/2017-12-31_17-46-32/adolescent#006.csv b/examples/results/2017-12-31_17-46-32/adolescent#006.csv
diff --git a/examples/results/2017-12-31_17-46-32/adolescent#007.csv b/examples/results/2017-12-31_17-46-32/adolescent#007.csv
diff --git a/examples/results/2017-12-31_17-46-32/adolescent#008.csv b/examples/results/2017-12-31_17-46-32/adolescent#008.csv
diff --git a/examples/results/2017-12-31_17-46-32/adolescent#009.csv b/examples/results/2017-12-31_17-46-32/adolescent#009.csv
diff --git a/examples/results/2017-12-31_17-46-32/adolescent#010.csv b/examples/results/2017-12-31_17-46-32/adolescent#010.csv
diff --git a/examples/results/2017-12-31_17-46-32/adult#001.csv b/examples/results/2017-12-31_17-46-32/adult#001.csv
diff --git a/examples/results/2017-12-31_17-46-32/adult#002.csv b/examples/results/2017-12-31_17-46-32/adult#002.csv
diff --git a/examples/results/2017-12-31_17-46-32/adult#003.csv b/examples/results/2017-12-31_17-46-32/adult#003.csv
diff --git a/examples/results/2017-12-31_17-46-32/adult#004.csv b/examples/results/2017-12-31_17-46-32/adult#004.csv
diff --git a/examples/results/2017-12-31_17-46-32/adult#005.csv b/examples/results/2017-12-31_17-46-32/adult#005.csv
diff --git a/examples/results/2017-12-31_17-46-32/adult#006.csv b/examples/results/2017-12-31_17-46-32/adult#006.csv
diff --git a/examples/results/2017-12-31_17-46-32/adult#007.csv b/examples/results/2017-12-31_17-46-32/adult#007.csv
diff --git a/examples/results/2017-12-31_17-46-32/adult#008.csv b/examples/results/2017-12-31_17-46-32/adult#008.csv
diff --git a/examples/results/2017-12-31_17-46-32/adult#009.csv b/examples/results/2017-12-31_17-46-32/adult#009.csv
diff --git a/examples/results/2017-12-31_17-46-32/adult#010.csv b/examples/results/2017-12-31_17-46-32/adult#010.csv
diff --git a/examples/results/2017-12-31_17-46-32/child#001.csv b/examples/results/2017-12-31_17-46-32/child#001.csv
diff --git a/examples/results/2017-12-31_17-46-32/child#002.csv b/examples/results/2017-12-31_17-46-32/child#002.csv
diff --git a/examples/results/2017-12-31_17-46-32/child#003.csv b/examples/results/2017-12-31_17-46-32/child#003.csv
diff --git a/examples/results/2017-12-31_17-46-32/child#004.csv b/examples/results/2017-12-31_17-46-32/child#004.csv
diff --git a/examples/results/2017-12-31_17-46-32/child#005.csv b/examples/results/2017-12-31_17-46-32/child#005.csv
diff --git a/examples/results/2017-12-31_17-46-32/child#006.csv b/examples/results/2017-12-31_17-46-32/child#006.csv
diff --git a/examples/results/2017-12-31_17-46-32/child#007.csv b/examples/results/2017-12-31_17-46-32/child#007.csv
diff --git a/examples/results/2017-12-31_17-46-32/child#008.csv b/examples/results/2017-12-31_17-46-32/child#008.csv
diff --git a/examples/results/2017-12-31_17-46-32/child#009.csv b/examples/results/2017-12-31_17-46-32/child#009.csv
diff --git a/examples/results/2017-12-31_17-46-32/child#010.csv b/examples/results/2017-12-31_17-46-32/child#010.csv
diff --git a/examples/results/2017-12-31_17-46-32/performance_stats.csv b/examples/results/2017-12-31_17-46-32/performance_stats.csv
diff --git a/examples/results/2017-12-31_17-46-32/risk_stats.png b/examples/results/2017-12-31_17-46-32/risk_stats.png
diff --git a/examples/results/2017-12-31_17-46-32/risk_trace.csv b/examples/results/2017-12-31_17-46-32/risk_trace.csv
diff --git a/examples/results/2017-12-31_17-46-32/zone_stats.png b/examples/results/2017-12-31_17-46-32/zone_stats.png
diff --git a/examples/run_gymnasium.py b/examples/run_gymnasium.py
@@ -4,7 +4,7 @@
 register(
     id="simglucose/adolescent2-v0",
     entry_point="simglucose.envs:T1DSimGymnaisumEnv",
-    max_episode_steps=10,
+    max_episode_steps=1000,
     kwargs={"patient_name": "adolescent#002"},
 )
 

diff --git a/examples/run_mpc.py b/examples/run_mpc.py
@@ -0,0 +1,42 @@
+import pandas as pd
+from datetime import timedelta, datetime
+from simglucose.controller.mpc_ctrller import PatientModel, MPCController
+from simglucose.simulation.user_interface import simulate
+from simglucose.simulation.scenario_gen import RandomScenario
+
+RESULT_PATH = './results/sysid_results/adolescent#001.csv'
+df_result = pd.read_csv(RESULT_PATH)
+df_result = df_result.iloc[:-1, :]
+patient_name = 'adolescent#001'
+na = 3
+nb = 5
+nc = 5
+X = df_result.loc[:, ['insulin', 'CHO']].values
+y = df_result['CGM'].values
+mdl = PatientModel(patient_name, na, nb, nc, X, y)
+
+# print(mdl.params)
+# print(mdl.alpha)
+# print(mdl.beta)
+
+N = 60  # horizon 5hr
+Q = 1  #
+QN = 10  # final cost
+R = 1000  # input penality
+
+ctrller = MPCController(patient_name, mdl, N, Q, QN, R, insulin_max=1)
+
+SIM_RESULT_PATH = './results/mpc_results'
+start_time = datetime.combine(datetime.now().date(),
+                              datetime.min.time()) + timedelta(hours=7)
+simulate(sim_time=timedelta(hours=24),
+         scenario=RandomScenario(start_time=start_time, seed=1),
+         patient_names=['adolescent#001'],
+         cgm_name='Dexcom',
+         cgm_seed=1,
+         insulin_pump_name='Insulet',
+         controller=ctrller,
+         start_time=start_time,
+         save_path=SIM_RESULT_PATH,
+         animate=True,
+         parallel=False)
diff --git a/examples/run_sys_id_experiments.py b/examples/run_sys_id_experiments.py
@@ -0,0 +1,22 @@
+from datetime import timedelta, datetime
+from simglucose.controller.sys_id_controller import SysIDController
+from simglucose.simulation.user_interface import simulate
+from simglucose.simulation.scenario import CustomScenario
+
+RESULT_PATH = './results/sysid_results'
+
+start_time = datetime.combine(datetime.now().date(), datetime.min.time())
+
+insulin_time = [start_time + timedelta(hours=10)]
+insulin_amount = [0.5]
+sysid_controller = SysIDController(insulin_time, insulin_amount)
+
+simulate(sim_time=timedelta(hours=24),
+         scenario=CustomScenario(start_time=start_time,
+                                 scenario=[(start_time + timedelta(hours=7),
+                                            30)]),
+         controller=sysid_controller,
+         start_time=start_time,
+         save_path=RESULT_PATH,
+         animate=True,
+         parallel=True)
diff --git a/results/sysid_results/BG_trace.png b/results/sysid_results/BG_trace.png
diff --git a/results/sysid_results/CVGA.png b/results/sysid_results/CVGA.png
diff --git a/results/sysid_results/CVGA_stats.csv b/results/sysid_results/CVGA_stats.csv
@@ -0,0 +1,2 @@
+,A,B,C,D,E
+0,0.0,0.2,0.03333333333333333,0.16666666666666666,0.03333333333333333