diff --git a/cami_src/evaluation_scripts/seed_variation_script.py b/cami_src/evaluation_scripts/seed_variation_script.py index baa420c831ef45e4cb757a187a22e312efff5ed3..5fdca37fb0d6ca4e70d19de9928dac8ccb58b8ad 100644 --- a/cami_src/evaluation_scripts/seed_variation_script.py +++ b/cami_src/evaluation_scripts/seed_variation_script.py @@ -52,13 +52,13 @@ def make_seedvariation(cami, n_iterations, removal_frac=0.2, vis=False, plot=Fal for tool in prediction_tools: res_table.write(f'\t{tool}_msr_ks_pvalue') - # with open(os.path.join(cami.tmp_dir, f'{used_tools[0]}_{cami.uid}_relevance_scores.tsv'), 'r') as f: - # for line in f: - # val_name = line.split('\t')[0] - # redisc_table.write(f'\t{val_name}') - # res_table.write('\n') - # result dictionaries of the form {tool:list(value for each iteration)} + with open(os.path.join(cami.tmp_dir, f'{used_tools[0]}_{cami.uid}_relevance_scores.tsv'), 'r') as f: + for line in f: + val_name = line.split('\t')[0] + redisc_table.write(f'\t{val_name}') + res_table.write('\n') + #result dictionaries of the form {tool:list(value for each iteration)} tp_rate_dict = {k:list() for k in used_tools} redisc_rate_dict = {k:list() for k in used_tools} module_size_dict = {k:list() for k in used_tools} @@ -142,10 +142,10 @@ def make_seedvariation(cami, n_iterations, removal_frac=0.2, vis=False, plot=Fal list(module_size_dict[pred_tool])) res_table.write(f'\t{p_val}') - # with open(os.path.join(cami.tmp_dir, f'{tool}_{identifier}_relevance_scores.tsv'), 'r') as f: - # for line in f: - # rel_score = line.split('\t')[1].strip() - # res_table.write(f'\t{rel_score}') + with open(os.path.join(cami.tmp_dir, f'{tool}_{identifier}_relevance_scores.tsv'), 'r') as f: + for line in f: + rel_score = line.split('\t')[1].strip() + res_table.write(f'\t{rel_score}') res_table.write('\n') print(f'Result tables are saved in the following locations:')