Resolve "integration-sota"

Closes #15 (closed)

code/bolsonaro/visualization/plotter.py

@@ -51,6 +51,7 @@ class Plotter(object):
 
     @staticmethod
     def plot_mean_and_CI(ax, mean, lb, ub, x_value, color_mean=None, facecolor=None, label=None):
+        #print(x_value, mean, lb, ub)
         # plot the shaded range of the confidence intervals
         ax.fill_between(x_value, ub, lb, facecolor=facecolor, alpha=.5)
         # plot the mean on top
@@ -105,7 +106,7 @@ class Plotter(object):
 
     @staticmethod
     def plot_stage2_losses(file_path, all_experiment_scores, x_value,
-        xlabel, ylabel, all_labels, title):
+        xlabel, ylabel, all_labels, title, filter_num=-1):
 
         fig, ax = plt.subplots()
 
@@ -124,13 +125,14 @@ class Plotter(object):
             # Compute the mean and the std for the CI
             mean_experiment_scores = np.average(experiment_scores, axis=0)
             std_experiment_scores = np.std(experiment_scores, axis=0)
+
             # Plot the score curve with the CI
             Plotter.plot_mean_and_CI(
                 ax=ax,
                 mean=mean_experiment_scores,
                 lb=mean_experiment_scores + std_experiment_scores,
                 ub=mean_experiment_scores - std_experiment_scores,
-                x_value=x_value,
+                x_value=x_value[:filter_num] if len(mean_experiment_scores) == filter_num else x_value,
                 color_mean=colors[i],
                 facecolor=colors[i],
                 label=all_labels[i]