diff --git a/figures/tuning_comparison.py b/figures/tuning_comparison.py index cb24dea..5d0f267 100644 --- a/figures/tuning_comparison.py +++ b/figures/tuning_comparison.py @@ -97,8 +97,9 @@ def plot(args, bmark, inefficiency, nt_points_list, t_points_list): nt_ineff = [x['inefficiency'] for x in nt_points] t_ineff = [x['inefficiency'] for x in t_points] - ax.plot(x_axis, nt_ineff, 'k') - ax.plot(x_axis, t_ineff) + ax.plot(x_axis, nt_ineff, 'k', label='No Tuning') + ax.plot(x_axis, t_ineff, label='W. Tuning') + ax.legend() ax = axes_array[1] @@ -130,7 +131,7 @@ def plot(args, bmark, inefficiency, nt_points_list, t_points_list): out_fname = os.path.join(os.path.join(os.path.join(args.output_dir, 'tuning_comparison'), bmark), '%.1f.jpg' % (inefficiency)) if not os.path.exists(os.path.dirname(out_fname)): os.makedirs(os.path.dirname(out_fname)) - plt.savefig(out_fname, dpi=300) + plt.savefig(out_fname) def main(argv): args = parse(argv)