Browse Source

add a startup event for loading q learning model

chenhaiyang 4 years ago
parent
commit
33c80a0970
1 changed files with 15 additions and 0 deletions
  1. 15 0
      app/controllers/events.py

+ 15 - 0
app/controllers/events.py

@@ -0,0 +1,15 @@
+# -*- coding: utf-8 -*-
+
+import scipy.io as sio
+
+from app.core.config import settings
+
+q_learning_models = {}
+
+
+async def load_q_learning_model():
+    base_path = settings.PROJECT_DIR
+    summer_model_path = f'{base_path}/app/resources/ml_models/equipment/fcu/q_learning/net_1_summer.mat'
+    winter_model_path = f'{base_path}/app/resources/ml_models/equipment/fcu/q_learning/net_1_winter.mat'
+    q_learning_models.update({'summer': sio.loadmat(summer_model_path)['net'][0, 0][0]})
+    q_learning_models.update({'winter': sio.loadmat(winter_model_path)['net'][0, 0][0]})