import yaml def load_config(): with open('./config/database_config.yaml') as file: config = yaml.safe_load(file) return config def load_model_config(): with open('./config/model_config.yaml') as file: config = yaml.safe_load(file) return config