Mercurial > pylearn

view cost.py @ 419:43d9aa93934e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added other_ops.py to nnet_ops; added basic tests, no docs.
author James Bergstra <bergstrj@iro.umontreal.ca>
date Mon, 14 Jul 2008 16:48:02 -0400
parents 319bf28c2dd5
children 0f366ecb11ee
line wrap: on
line source

"""
Cost functions.
"""

import theano.tensor as T

def quadratic(target, output, axis=1):
    return T.mean(T.sqr(target - output), axis)

def cross_entropy(target, output, axis=1):
    return -T.mean(target * T.log2(output) + (1 - target) * T.log2(1 - output), axis=axis)