elif which_dataset == 'saapdb':
    mutation_to_class = helper.saapdb_to_class
elif which_dataset == 'p53':
    mutation_to_class = helper.p53_to_class
elif which_dataset == 'cbs':
    mutation_to_class = helper.cbs_to_class
elif which_dataset == 'p53_stone':
    mutation_to_class = helper.p53_stone_to_class
elif which_dataset == 'lys_stone':
    mutation_to_class = helper.lys_stone_to_class
elif which_dataset == 'reverse_stone':
    mutation_to_class = helper.reverse_stone_to_class
elif which_dataset == 'protease_stone':
    mutation_to_class = helper.protease_stone_to_class
elif which_dataset == 'hemo_stone':
    mutation_to_class = helper.hemo_stone_to_class
my_output = objects.get_output_obj(params, my_l, use_neighbor, ignore_pos, max_neighbor, num_trials, pseudo_total, helper.vanilla_similarity, helper.normalize_nothing, mutation_to_class, to_neighbor_p_value)

comm.Barrier()

root_output = comm.gather(my_output, root=0)

to_use = []


if rank == 0:
    for output in root_output:
        to_use = to_use + output

        helper.write_mat_raw(to_use, output_file)
Exemplo n.º 2
0
 def dump_object(self, obj):
     helper.write_mat_raw(obj, self.get_holding_location())