def compute_logz(rbm_params):
(nvis, nhid) = rbm_params[0].shape
model = rbm.RBM(nvis, nhid)
model.weights.set_value(rbm_params[0])
model.visbias.set_value(rbm_params[1])
model.hidbias.set_value(rbm_params[2])
hid = T.matrix('hid')
hid_fe = model.free_energy_given_h(hid)
free_energy_fn = theano.function([hid], hid_fe)
return rbm_tools.compute_log_z(model, free_energy_fn)
def compute_logz(rbm_params):
(nvis, nhid) = rbm_params[0].shape
model = rbm.RBM(nvis, nhid)
model.transformer.get_params()[0].set_value(rbm_params[0])
model.bias_vis.set_value(rbm_params[1])
model.bias_hid.set_value(rbm_params[2])
hid = T.matrix('hid')
hid_fe = model.free_energy_given_h(hid)
free_energy_fn = theano.function([hid], hid_fe)
return rbm_tools.compute_log_z(model, free_energy_fn)