Mercurial > pylearn
annotate test_dataset.py @ 51:59757365a057
the script can be autorun
| author | Frederic Bastien <bastienf@iro.umontreal.ca> |
|---|---|
| date | Tue, 29 Apr 2008 14:40:33 -0400 |
| parents | a5c70dc42972 |
| children | e3ac93e27e16 |
| rev | line source |
|---|---|
|
51
59757365a057
the script can be autorun
Frederic Bastien <bastienf@iro.umontreal.ca>
parents:
45
diff
changeset
|
1 #!/bin/env python |
| 45 | 2 from dataset import * |
| 3 from math import * | |
| 4 import numpy | |
| 5 | |
| 6 def test1(): | |
| 7 global a,ds | |
| 8 a = numpy.random.rand(10,4) | |
| 9 print a | |
| 10 ds = ArrayDataSet(a,{'x':slice(3),'y':3,'z':[0,2]}) | |
| 11 print "len(ds)=",len(ds) | |
| 12 print "example 0 = ",ds[0] | |
| 13 print "x=",ds["x"] | |
| 14 print "x|y" | |
| 15 for x,y in ds("x","y"): | |
| 16 print x,y | |
| 17 minibatch_iterator = ds.minibatches(fieldnames=['z','y'],n_batches=1,minibatch_size=3,offset=4) | |
| 18 minibatch = minibatch_iterator.__iter__().next() | |
| 19 print "minibatch=",minibatch | |
| 20 for var in minibatch: | |
| 21 print "var=",var | |
| 22 print "take a slice:",ds[1:6:2] | |
| 23 | |
| 24 test1() |