minor fix for plotting testresult

automl · Feb 12, 2015 · d6c5e3b · d6c5e3b
1 parent 94e0145
commit d6c5e3b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/HPOlib/Plotting/plotTrace_perEval.py b/HPOlib/Plotting/plotTrace_perEval.py
@@ -76,6 +76,9 @@ def main(pkl_list, name_list, autofill, optimum=0, save="", title="", log=False,
                                                      test=False)
             trial_list[-1].append(np.array(trace))
 
+    if not plot_test_performance:
+        test_list = None
+
     for i in range(len(trial_list)):
         max_len = max([len(ls) for ls in trial_list[i]])
         for t in range(len(trial_list[i])):

diff --git a/HPOlib/Plotting/plot_trajectory.py b/HPOlib/Plotting/plot_trajectory.py
@@ -136,7 +136,6 @@ def plot_trajectories(trial_list, name_list, x_ticks,
                  marker=marker, markersize=properties["markersize"],
                  label=label)
         if test_trials is not None:
-            print i, test_trials[i][:][0], test_trials[i][:][1]
             ax1.scatter(x=test_trials[i][:][0], y=test_trials[i][:][1],
                         color=color, marker='o', s=20)