Spaces:
Running
Running
Update src/vis_utils.py
Browse files- 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(
|
| 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(
|
| 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)
|