events.py 572 B

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