comparison test_dataset.py @ 56:1729ad44f175

Automated merge with ssh://p-omega1@lgcm.iro.umontreal.ca/tlearn
author Yoshua Bengio <bengioy@iro.umontreal.ca>
date Tue, 29 Apr 2008 16:09:17 -0400
parents 70147d00615a
children 17729d7104fa
comparison
equal deleted inserted replaced
55:66619ce44497 56:1729ad44f175
1 1 #!/bin/env python
2 from dataset import * 2 from dataset import *
3 from math import * 3 from math import *
4 import numpy 4 import numpy
5 5
6 def test1(): 6 def test1():
7 global a,ds 7 global a,ds
8 a = numpy.random.rand(10,4) 8 a = numpy.random.rand(10,4)
9 print a 9 print a
10 ds = ArrayDataSet(a,{'x':slice(3),'y':3,'z':[0,2]}) 10 ds = ArrayDataSet(a,{'x':slice(3),'y':3,'z':[0,2]})
11 print "len(ds)=",len(ds) 11 print "len(ds)=",len(ds)
12 assert(len(ds)==10)
12 print "example 0 = ",ds[0] 13 print "example 0 = ",ds[0]
14 # assert
13 print "x=",ds["x"] 15 print "x=",ds["x"]
14 print "x|y" 16 print "x|y"
15 for x,y in ds("x","y"): 17 for x,y in ds("x","y"):
16 print x,y 18 print x,y
17 minibatch_iterator = ds.minibatches(fieldnames=['z','y'],n_batches=1,minibatch_size=3,offset=4) 19 minibatch_iterator = ds.minibatches(fieldnames=['z','y'],n_batches=1,minibatch_size=3,offset=4)