Mercurial > pylearn
annotate test_dataset.py @ 48:b6730f9a336d
Fixing MinibatchDataSet getitem
| author | bengioy@grenat.iro.umontreal.ca |
|---|---|
| date | Tue, 29 Apr 2008 13:40:13 -0400 |
| parents | a5c70dc42972 |
| children | e3ac93e27e16 |
| rev | line source |
|---|---|
| 45 | 1 |
| 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 | |
|
48
b6730f9a336d
Fixing MinibatchDataSet getitem
bengioy@grenat.iro.umontreal.ca
parents:
45
diff
changeset
|
22 print "take a slice and look at field y",ds[1:6:2]["y"] |
| 45 | 23 |
| 24 test1() |