comparison pylearn/algorithms/mcRBM.py @ 1351:6402b3309ece

mcRBM - added to docstring
author James Bergstra <bergstrj@iro.umontreal.ca>
date Thu, 28 Oct 2010 16:15:47 -0400
parents d957155264da
children 54b2268db0d7
comparison
equal deleted inserted replaced
1350:d957155264da 1351:6402b3309ece
270 :param free_energy_fn: lambda (TensorType matrix MxN) -> TensorType vector of M free energies 270 :param free_energy_fn: lambda (TensorType matrix MxN) -> TensorType vector of M free energies
271 :param pos_v: positive-phase sample of visible units 271 :param pos_v: positive-phase sample of visible units
272 :param neg_v: negative-phase sample of visible units 272 :param neg_v: negative-phase sample of visible units
273 :param wrt: TensorType variables with respect to which we want gradients (similar to the 273 :param wrt: TensorType variables with respect to which we want gradients (similar to the
274 'wrt' argument to tensor.grad) 274 'wrt' argument to tensor.grad)
275 :param other_cost: TensorType scalar 275 :param other_cost: TensorType scalar (should be the sum over a minibatch, not mean)
276 276
277 :returns: TensorType variables for the gradient on each of the 'wrt' arguments 277 :returns: TensorType variables for the gradient on each of the 'wrt' arguments
278 278
279 279
280 :math: Cost = other_cost + \sum_i free_energy(pos_v[i]) - free_energy(neg_v[i]) 280 :math: Cost = other_cost + \sum_i free_energy(pos_v[i]) - free_energy(neg_v[i])