|
356
|
1 from onehotop import one_hot
|
|
|
2
|
|
|
3 import unittest
|
|
|
4 from theano import compile
|
|
|
5 from theano import gradient
|
|
|
6
|
|
|
7 from theano.tensor import as_tensor
|
|
|
8
|
|
|
9 import random
|
|
|
10 import numpy.random
|
|
|
11
|
|
|
12 class T_OneHot(unittest.TestCase):
|
|
|
13 def test0(self):
|
|
|
14 x = as_tensor([3, 2, 1])
|
|
|
15 y = as_tensor(5)
|
|
|
16 o = one_hot(x, y)
|
|
|
17 y = compile.eval_outputs([o])
|
|
|
18 self.failUnless(numpy.all(y == numpy.asarray([[0, 0, 0, 1, 0], [0, 0, 1, 0, 0], [0, 1, 0, 0, 0]])))
|
|
|
19
|
|
|
20 if __name__ == '__main__':
|
|
|
21 unittest.main()
|
