|
24
|
1
|
|
|
2 import unittest
|
|
|
3 import theano._test_tensor as TT
|
|
|
4 import numpy
|
|
|
5
|
|
|
6 from nnet_ops import *
|
|
|
7
|
|
|
8 class T_sigmoid(unittest.TestCase):
|
|
|
9 def setUp(self):
|
|
|
10 numpy.random.seed(9999)
|
|
|
11 def test_elemwise(self):
|
|
|
12 TT.verify_grad(self, Sigmoid, [numpy.random.rand(3,4)])
|
|
|
13
|
|
|
14
|
|
|
15 class T_CrossentropySoftmax1Hot(unittest.TestCase):
|
|
|
16 def setUp(self):
|
|
|
17 numpy.random.seed(9999)
|
|
|
18 def test0(self):
|
|
|
19 y_idx = [0,1,3]
|
|
|
20 def output1(a):
|
|
|
21 return cross_entropy_softmax_1hot(a, y_idx)[0:1]
|
|
|
22 TT.verify_grad(self, output1, [numpy.random.rand(3,4)])
|
|
|
23
|
|
|
24
|
|
|
25
|
|
|
26 if __name__ == '__main__':
|
|
|
27 unittest.main()
|
