config.py 288 B

1234567891011
  1. import yaml
  2. def load_config():
  3. with open('./config/database_config.yaml') as file:
  4. config = yaml.safe_load(file)
  5. return config
  6. def load_model_config():
  7. with open('./config/model_config.yaml') as file:
  8. config = yaml.safe_load(file)
  9. return config