ssrbf bug fix

This commit is contained in:
Zhenwen Dai 2014-11-13 11:44:01 +00:00
parent 2724e7d476
commit bee1bc360e

View file

@ -177,8 +177,8 @@ class SpikeAndSlabPosterior(VariationalPosterior):
def gamma_log_prob(self):
loggamma = param_to_array(self.gamma).copy()
loggamma[loggamma>-40] = -np.log1p(np.exp(-loggamma[loggamma>-40]))
loggamma1 = param_to_array(self.gamma).copy()
loggamma1[loggamma1<40] = -np.log1p(np.exp(loggamma1[loggamma1<40]))
loggamma1 = -param_to_array(self.gamma).copy()
loggamma1[loggamma1>-40] = -np.log1p(np.exp(-loggamma1[loggamma1>-40]))
return loggamma,loggamma1
def set_gradients(self, grad):