Spaces:
Sleeping
Sleeping
| import sys | |
| import yaml | |
| import pandas as pd | |
| from utils import * | |
| def load_config(filename): | |
| with open(filename, 'r') as stream: | |
| try: | |
| return yaml.safe_load(stream) | |
| except yaml.YAMLError as exc: | |
| print(exc) | |
| def save_results(dsm_result, spm_result, occurrence_matrix, dsm_result_path, spm_result_path, occurrence_matrix_path): | |
| dsm_result.to_csv(dsm_result_path, index=False) | |
| spm_result.to_csv(spm_result_path, index=False) | |
| occurrence_matrix.to_csv(occurrence_matrix_path, index=False) | |
| def main(): | |
| if len(sys.argv) > 1: | |
| config_file = 'config.yaml' | |
| command_args = sys.argv[1:] | |
| overridden_params = {} | |
| for arg in command_args: | |
| if '=' in arg: | |
| key, value = arg.split('=') | |
| overridden_params[key] = value | |
| else: | |
| print(f"Ignoring invalid argument: {arg}") | |
| config = load_config(config_file) | |
| # SPM parameters | |
| spm_params = config.get('SPM', {}) | |
| for key, value in overridden_params.items(): | |
| if key in spm_params: | |
| spm_params[key] = value | |
| spm_result, occurrence_matrix = SPM(spm_params) | |
| # DSM parameters | |
| dsm_params = config.get('DSM', {}) | |
| for key, value in overridden_params.items(): | |
| if key in dsm_params: | |
| dsm_params[key] = value | |
| ptrn_left, ptrn_right, ptrn_both_left, ptrn_both_right, dsm_result = DSM(dsm_params) | |
| # Fetching paths from config | |
| dsm_result_path = config.get('dsm_result_path') | |
| spm_result_path = config.get('spm_result_path') | |
| occurrence_matrix_path = config.get('occurrence_matrix_path') | |
| # Saving results | |
| save_results(dsm_result, spm_result, occurrence_matrix, dsm_result_path, spm_result_path, occurrence_matrix_path) | |
| else: | |
| config_file = 'config.yaml' | |
| config = load_config(config_file) | |
| # SPM parameters | |
| spm_params = config.get('SPM', {}) | |
| spm_result, occurrence_matrix = SPM(spm_params) | |
| # DSM parameters | |
| dsm_params = config.get('DSM', {}) | |
| ptrn_left, ptrn_right, ptrn_both_left, ptrn_both_right, dsm_result = DSM(dsm_params) | |
| # Fetching paths from config | |
| dsm_result_path = config.get('dsm_result_path') | |
| spm_result_path = config.get('spm_result_path') | |
| occurrence_matrix_path = config.get('occurrence_matrix_path') | |
| # Saving results | |
| save_results(dsm_result, spm_result, occurrence_matrix, dsm_result_path, spm_result_path, occurrence_matrix_path) | |
| if __name__ == "__main__": | |
| main() | |