Spaces:
Sleeping
Sleeping
Update src/vis_utils.py
Browse files- src/vis_utils.py +3 -3
src/vis_utils.py
CHANGED
@@ -170,7 +170,7 @@ def plot_function_results(method_names, aspect, metric, function_path="/tmp/func
|
|
170 |
|
171 |
return filename
|
172 |
|
173 |
-
def plot_family_results(
|
174 |
if not os.path.exists(family_path):
|
175 |
benchmark_types = ["similarity", "function", "family", "affinity"] #download all files for faster results later
|
176 |
download_from_hub(benchmark_types)
|
@@ -246,8 +246,8 @@ def plot_affinity_results(method_names, metric, affinity_path="/tmp/affinity_res
|
|
246 |
|
247 |
# Customize x-axis and y-axis tickers and grid
|
248 |
ax.xaxis.set_major_locator(ticker.MultipleLocator(5))
|
249 |
-
ax.get_xaxis().set_minor_locator(
|
250 |
-
ax.get_yaxis().set_minor_locator(
|
251 |
ax.grid(b=True, which='major', color='gainsboro', linewidth=1.0)
|
252 |
ax.grid(b=True, which='minor', color='whitesmoke', linewidth=0.5)
|
253 |
|
|
|
170 |
|
171 |
return filename
|
172 |
|
173 |
+
def plot_family_results(method_names, dataset, metric, family_path="/tmp/family_results.csv"):
|
174 |
if not os.path.exists(family_path):
|
175 |
benchmark_types = ["similarity", "function", "family", "affinity"] #download all files for faster results later
|
176 |
download_from_hub(benchmark_types)
|
|
|
246 |
|
247 |
# Customize x-axis and y-axis tickers and grid
|
248 |
ax.xaxis.set_major_locator(ticker.MultipleLocator(5))
|
249 |
+
ax.get_xaxis().set_minor_locator(ticker.AutoMinorLocator())
|
250 |
+
ax.get_yaxis().set_minor_locator(ticker.AutoMinorLocator())
|
251 |
ax.grid(b=True, which='major', color='gainsboro', linewidth=1.0)
|
252 |
ax.grid(b=True, which='minor', color='whitesmoke', linewidth=0.5)
|
253 |
|