Mercurial > pylearn

comparison _test_onehotop.py @ 356:18702ceb2096

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Added more functions
author Joseph Turian <turian@iro.umontreal.ca>
date Thu, 19 Jun 2008 16:18:37 -0400
parents
children 844bad76459c
comparison
equal deleted inserted replaced
355:430c9e92cd23 356:18702ceb2096
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()