Commit f84f8120 authored by Martino Bertoni's avatar Martino Bertoni 🌋
Browse files

re-enabling plots for adanet results

parent 02d1db1c
...@@ -443,7 +443,7 @@ class AdaNetWrapper(object): ...@@ -443,7 +443,7 @@ class AdaNetWrapper(object):
return model_vars return model_vars
def save_performances(self, output_dir, plot, def save_performances(self, output_dir, plot,
suffix=None, extra_predictors=None, do_plot=False): suffix=None, extra_predictors=None, do_plot=True):
"""Save stats and make plots.""" """Save stats and make plots."""
# read input # read input
splits = ['train', 'test', 'validation'] splits = ['train', 'test', 'validation']
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment