Mercurial > pylearn
view test_dataset.py @ 55:66619ce44497
Efficient implementation of getitem for ArrayDataSet
author | Yoshua Bengio <bengioy@iro.umontreal.ca> |
---|---|
date | Tue, 29 Apr 2008 15:05:12 -0400 |
parents | b6730f9a336d |
children | e3ac93e27e16 |
line wrap: on
line source
from dataset import * from math import * import numpy def test1(): global a,ds a = numpy.random.rand(10,4) print a ds = ArrayDataSet(a,{'x':slice(3),'y':3,'z':[0,2]}) print "len(ds)=",len(ds) print "example 0 = ",ds[0] print "x=",ds["x"] print "x|y" for x,y in ds("x","y"): print x,y minibatch_iterator = ds.minibatches(fieldnames=['z','y'],n_batches=1,minibatch_size=3,offset=4) minibatch = minibatch_iterator.__iter__().next() print "minibatch=",minibatch for var in minibatch: print "var=",var print "take a slice and look at field y",ds[1:6:2]["y"] test1()