Merge pull request JasperSnoek#52 from emcastillo/master

1 variable with 1 dimension fixed
wuda01 · Feb 12, 2016 · b37a541 · b37a541
2 parents 9fd9a4a + 70309f0
commit b37a541
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/spearmint/spearmint/chooser/GPEIOptChooser.py b/spearmint/spearmint/chooser/GPEIOptChooser.py
@@ -508,7 +508,11 @@ def grad_optimize_ei(self, cand, comp, pend, vals, compute_grad=True):
             g_ei_s2 = 0.5*npdf / func_s
 
             # Apply covariance function
-            grad_cross = np.squeeze(cand_cross_grad)
+            # Squeeze can break the 1D case be careful
+            if pend.shape[1] == 1:
+                grad_cross = np.squeeze(cand_cross_grad, axis=(2,))
+            else:
+                grad_cross = np.squeeze(cand_cross_grad)
 
             grad_xp_m = np.dot(alpha.transpose(),grad_cross)
             grad_xp_v = np.dot(-2*spla.cho_solve(