Source Code for Module QSTK.qstklearn.gendata

 1  import random 
 2 -def gendata(N,d,bounds,clsses,fname): 
 3          outf = open(fname,'w') 
 4          for i in xrange(N): 
 5                  pnt = [None,]*(d+1) 
 6                  for x in xrange(d): 
 7                          pnt[x] = random.uniform(bounds[x][0],bounds[x][1]) 
 8                  pnt[d] = random.choice(clsses) 
 9                  outf.write(", ".join(map(str,pnt))+"\n") 
10          outf.close() 
11 -def gensingle(d,bounds,clsses): 
12          pnt = [None,]*(d+1) 
13          for x in xrange(d): 
14                  pnt[x] = random.uniform(bounds[x][0],bounds[x][1]) 
15          pnt[d] = random.choice(clsses) 
16          return pnt 
17