File size: 1,522 Bytes
9157432 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import json
import os
from pathlib import Path
from easydict import EasyDict as edict
def get_config_regression(model_name, dataset_name, config_file=""):
"""
Get the regression config of given dataset and model from config file.
Parameters:
config_file (str): Path to config file, if given an empty string, will use default config file.
model_name (str): Name of model.
dataset_name (str): Name of dataset.
Returns:
config (dict): config of the given dataset and model
"""
if config_file == "":
config_file = Path(__file__).parent / "config" / "config_regression.json"
with open(config_file, 'r') as f:
config_all = json.load(f)
model_common_args = config_all[model_name]['commonParams']
model_dataset_args = config_all[model_name]['datasetParams'][dataset_name]
dataset_args = config_all['datasetCommonParams'][dataset_name]
# use aligned feature if the model requires it, otherwise use unaligned feature
dataset_args = dataset_args['aligned'] if (model_common_args['need_data_aligned'] and 'aligned' in dataset_args) else dataset_args['unaligned']
config = {}
config['model_name'] = model_name
config['dataset_name'] = dataset_name
config.update(dataset_args)
config.update(model_common_args)
config.update(model_dataset_args)
config['featurePath'] = os.path.join(config_all['datasetCommonParams']['dataset_root_dir'], config['featurePath'])
config = edict(config)
return config
|