diff --git a/Swing/Swing.py b/Swing/Swing.py index 22180bf..8c99405 100644 --- a/Swing/Swing.py +++ b/Swing/Swing.py @@ -548,7 +548,7 @@ def compile_roller_edges(self, self_edges=False): #todo: returns nothing? return - def make_static_edge_dict(self, true_edges, self_edges=False, lag_method='max_median'): + def make_static_edge_dict(self, self_edges=False, lag_method='max_median'): """ Make a dictionary of edges :return: diff --git a/run_swing.py b/run_swing.py index 9c432b6..cc950f1 100644 --- a/run_swing.py +++ b/run_swing.py @@ -1,6 +1,5 @@ import pandas as pd from Swing import Swing -from Swing.util.Evaluator import Evaluator # Load data gene_start_column = 1 @@ -32,10 +31,7 @@ # Load the reference standard for scoring in silico predictions -evaluator = Evaluator(gold_standard_file, '\t') -true_edges = evaluator.gs_flat.tolist() - -sg.make_static_edge_dict(true_edges, self_edges=False, lag_method='mean_mean') +sg.make_static_edge_dict(self_edges=False, lag_method='mean_mean') ranked_edges = sg.make_sort_df(sg.edge_dict) roc_dict, pr_dict = sg.score(ranked_edges, gold_standard_file=gold_standard_file)