a more sensible prior for the amplitude

ktrapeznikov · May 6, 2014 · ff2d757 · ff2d757
1 parent 918653e
commit ff2d757
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spearmint/spearmint/chooser/GPEIOptChooser.py b/spearmint/spearmint/chooser/GPEIOptChooser.py
@@ -657,7 +657,7 @@ def logprob(hypers):
             lp += np.log(np.log(1 + (self.noise_scale/noise)**2))
 
             # Roll in amplitude lognormal prior
-            lp -= 0.5*(np.log(amp2)/self.amp2_scale)**2
+            lp -= 0.5*(np.log(np.sqrt(amp2))/self.amp2_scale)**2
 
             return lp
 
@@ -687,7 +687,7 @@ def logprob(hypers):
             lp    = -np.sum(np.log(np.diag(chol)))-0.5*np.dot(vals-mean, solve)
 
             # Roll in amplitude lognormal prior
-            lp -= 0.5*(np.log(amp2)/self.amp2_scale)**2
+            lp -= 0.5*(np.log(np.sqrt(amp2))/self.amp2_scale)**2
 
             return lp