|
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
|
|
|
22 print "take a slice:",ds[1:6:2]
|
|
|
23
|
|
|
24 test1()
|
