view _nnet_ops.py @ 67:810a8e3c85e1

fixed horrible memory leak from crossentropy...
author bergstra@is23.m
date Fri, 18 Apr 2008 03:35:58 -0400
parents bf0145fa73e8
children 8c2607f387e6
line wrap: on
line source


import unittest
import theano._test_tensor as TT
import numpy

from nnet_ops import *

class T_sigmoid(unittest.TestCase):
    def setUp(self):
        numpy.random.seed(9999)
    def test_elemwise(self):
        TT.verify_grad(self, Sigmoid, [numpy.random.rand(3,4)])


class T_CrossentropySoftmax1Hot(unittest.TestCase):
    def setUp(self):
        numpy.random.seed(9999)
    def test0(self):
        y_idx = [0,1,3]
        def output1(a,b):
            return crossentropy_softmax_1hot(a, b, y_idx)[0:1]
        TT.verify_grad(self, output1, [numpy.random.rand(3,4),
            numpy.random.rand(4)])



if __name__ == '__main__':
    unittest.main()