Spaces:
Running
Running
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 |
|