|
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] |
|
|
|
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 |
|
|
|
|
|
|
