Fix missing title in stage1 loss figure (still on the left; TODO: center the title).

code/bolsonaro/visualization/plotter.py

@@ -96,7 +96,7 @@ class Plotter(object):
         axes[0].set_xlabel(xlabel)
         axes[1].set_xlabel(xlabel)
         axes[0].set_ylabel(ylabel)
-        plt.suptitle(title)
+        axes[1].set_title(title)
         handles, labels = axes[0].get_legend_handles_labels()
         legend = axes[0].legend(handles, labels, loc='upper center', bbox_to_anchor=(1.1, -0.15))
         fig.savefig(file_path, dpi=fig.dpi, bbox_extra_artists=(legend,), bbox_inches='tight')

code/compute_results.py

@@ -192,7 +192,7 @@ if __name__ == "__main__":
             x_value=with_params_extracted_forest_sizes,
             xlabel='Number of trees extracted',
             ylabel='MSE', # TODO: hardcoded
-            title='Loss values of {} using best and default hyperparameters'.format(args.dataset_name)
+            title='Loss values of {}\nusing best and default hyperparameters'.format(args.dataset_name)
         )
     else:
         raise ValueError('This stage number is not supported yet, but it will be!')