Added legend to tuning_comparison.py

Guru Prasad
1 parent bd160bde
Showing 1 changed file with 4 additions and 3 deletions
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)