mgyigit commited on
Commit
9ee7f43
·
verified ·
1 Parent(s): 8e737a7

Update src/vis_utils.py

Browse files
Files changed (1) hide show
  1. 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(methods_selected, 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,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(mpl.ticker.AutoMinorLocator())
250
- ax.get_yaxis().set_minor_locator(mpl.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
 
 
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