mgyigit commited on
Commit
2bace10
·
verified ·
1 Parent(s): 81cc688

Update src/vis_utils.py

Browse files
Files changed (1) hide show
  1. src/vis_utils.py +2 -2
src/vis_utils.py CHANGED
@@ -26,7 +26,7 @@ def benchmark_plot(benchmark_type, methods_selected, x_metric, y_metric, aspect,
26
  if benchmark_type == 'similarity':
27
  return plot_similarity_results(methods_selected, x_metric, y_metric)
28
  elif benchmark_type == 'function':
29
- return plot_function_results("./data/function_results.csv", x_metric, y_metric, methods_selected)
30
  elif benchmark_type == 'family':
31
  return plot_family_results("./data/family_results.csv", methods_selected, x_metric, save_path="./plot_images")
32
  elif benchmark_type == "affinity":
@@ -119,7 +119,7 @@ def plot_similarity_results(methods_selected, x_metric, y_metric, similarity_pat
119
 
120
  return filename
121
 
122
- def plot_function_results(file_path, aspect, metric, method_names, function_path="/tmp/function_results.csv"):
123
  if not os.path.exists(function_path):
124
  benchmark_types = ["similarity", "function", "family", "affinity"] #download all files for faster results later
125
  download_from_hub(benchmark_types)
 
26
  if benchmark_type == 'similarity':
27
  return plot_similarity_results(methods_selected, x_metric, y_metric)
28
  elif benchmark_type == 'function':
29
+ return plot_function_results(apect, single_metric, methods_selected)
30
  elif benchmark_type == 'family':
31
  return plot_family_results("./data/family_results.csv", methods_selected, x_metric, save_path="./plot_images")
32
  elif benchmark_type == "affinity":
 
119
 
120
  return filename
121
 
122
+ def plot_function_results(aspect, metric, method_names, function_path="/tmp/function_results.csv"):
123
  if not os.path.exists(function_path):
124
  benchmark_types = ["similarity", "function", "family", "affinity"] #download all files for faster results later
125
  download_from_hub(benchmark_types)