コード例 #1
0
ファイル: test_lines.py プロジェクト: zoginni/helit
    row = docs[i * docGridWidth:(i + 1) * docGridWidth]
    rowExt = []
    for r in row:
        rowExt.append(r)
        rowExt.append(
            numpy.zeros((gridSpace, gridScale * 5), dtype=numpy.float_))
    rowExt = rowExt[:-1]
    rows.append(numpy.vstack(rowExt))

stack = []
for r in rows:
    stack.append(r)
    stack.append(numpy.zeros((r.shape[0], gridSpace), dtype=numpy.float_))
stack = stack[:-1]
docImage = numpy.hstack(stack).T * 255.0
img = cvarray.array2cv(docImage)
cv.SaveImage('test_lines/docs.png', img)

# Train...
params = dhdp.Params()
params.runs = 1
params.samples = 1
#params.burnIn = 10000
#c.setOneCluster(True)
#c.setCalcBeta(True)

print 'Fitting model...'
p = ProgBar()
model = c.sampleModel(params, p.callback)
del p
コード例 #2
0
        #print model.z.sum(axis=0)

        # Now plot the estimated distribution against the actual distribution...
        img = numpy.ones((height, width, 3))
        draw = model.sampleMixture()

        for px in xrange(width):
            x = float(px) / float(width) * (high - low) + low

            y_gt = 0.0
            for ii in xrange(len(gt)):
                y_gt += gt_weight[ii] * gt[ii].prob([x])
            y_gu = model.prob([x])
            y_gd = 0.0
            for ind, gauss in enumerate(draw[1]):
                y_gd += draw[0][ind] * gauss.prob([x])

            py_gt = int((1.0 - y_gt / scale) * height)
            py_gu = int((1.0 - y_gu / scale) * height)
            py_gd = numpy.clip(int((1.0 - y_gd / scale) * height), 0,
                               height - 1)

            img[py_gt, px, :] = [0.0, 1.0, 0.0]
            img[py_gu, px, :] = [1.0, 0.0, 0.0]
            img[py_gd, px, :] = [0.0, 0.0, 1.0]

        # Save plot out...
        img = cvarray.array2cv(img * 255.0)
        cv.SaveImage('%s/plot_%i.png' % (out_dir, i + 1), img)
        print
コード例 #3
0
ファイル: test_grid.py プロジェクト: zoginni/helit
    image *= 255.0 / image.max()
    image = numpy.reshape(image, (5, 5))
    image = numpy.repeat(numpy.repeat(image, 5, axis=0), 5, axis=1)
    image = numpy.append(image,
                         numpy.atleast_2d(numpy.zeros(image.shape[1])),
                         axis=0)
    image = numpy.append(image,
                         numpy.atleast_2d(numpy.zeros(image.shape[0])).T,
                         axis=1)
    docImageSet.append(image)

docVertSet = []
for i in xrange(50):
    docVertSet.append(numpy.vstack(docImageSet[i * 20:(i + 1) * 20]))
docSet = numpy.hstack(docVertSet)
img = cvarray.array2cv(docSet)
cv.SaveImage('test_grid_docs.png', img)

# Train...
print 'Trainning...'
#p = ProgBar()
#passes = vlda.solve()
#del p
passes = vlda.solveHuman()
print 'Took %i passes' % passes

# Generate an image of the final distributions associated with the learned documents...
# Get pixel values...
tImages = []
for topic in xrange(vlda.numTopics()):
    # Get distribution...
コード例 #4
0
  for px in xrange(width):
    x = float(px)/float(width) * (high-low) + low
     
    y_gt = 0.0
    for ii  in xrange(len(gt)):
      y_gt += gt_weight[ii] * gt[ii].prob([x])
    y_gu = model.prob([x])
    y_gd = 0.0
    for ind,gauss in enumerate(draw[1]):
      y_gd += draw[0][ind] * gauss.prob([x])
        
    py_gt = int((1.0 - y_gt/scale) * height)
    py_gu = int((1.0 - y_gu/scale) * height)
    py_gd = numpy.clip(int((1.0 - y_gd/scale) * height),0,height-1)

    img[py_gt,px,:] = [0.0,1.0,0.0]
    img[py_gu,px,:] = [1.0,0.0,0.0]
    img[py_gd,px,:] = [0.0,0.0,1.0]

  # Save plot out...
  img = cvarray.array2cv(img*255.0)
  cv.SaveImage('%s/plot_%i.png'%(out_dir,model.getStickCap()),img)
  print

  # Either finish or incriment the number of sticks for the next run...
  value = model.nllData()
  if prev==None or value<prev:
    prev = value
    model.incStickCap()
  else: break
コード例 #5
0
ファイル: test_grid.py プロジェクト: zerocolar/Project_Code
# Save out the input documents for confirmation (50x20 grid)...
docImageSet = []
for words in inputImageSet:
  image = numpy.asfarray(words)
  image *= 255.0/image.max()
  image = numpy.reshape(image,(5,5))
  image = numpy.repeat(numpy.repeat(image,5,axis=0),5,axis=1)
  image = numpy.append(image,numpy.atleast_2d(numpy.zeros(image.shape[1])),axis=0)
  image = numpy.append(image,numpy.atleast_2d(numpy.zeros(image.shape[0])).T,axis=1)
  docImageSet.append(image)

docVertSet = []
for i in xrange(50):
  docVertSet.append(numpy.vstack(docImageSet[i*20:(i+1)*20]))
docSet = numpy.hstack(docVertSet)
img = cvarray.array2cv(docSet)
cv.SaveImage('test_grid_docs.png',img)



# Train...
print 'Trainning...'
#p = ProgBar()
#passes = vlda.solve()
#del p
passes = vlda.solveHuman()
print 'Took %i passes'%passes



# Generate an image of the final distributions associated with the learned documents...
コード例 #6
0
ファイル: test_lines.py プロジェクト: PeterZhouSZ/helit
for i in xrange(documentsToTrain//docGridWidth):
  row = docs[i*docGridWidth:(i+1)*docGridWidth]
  rowExt = []
  for r in row:
    rowExt.append(r)
    rowExt.append(numpy.zeros((gridSpace,gridScale*5), dtype=numpy.float_))
  rowExt = rowExt[:-1]
  rows.append(numpy.vstack(rowExt))

stack = []
for r in rows:
  stack.append(r)
  stack.append(numpy.zeros((r.shape[0],gridSpace), dtype=numpy.float_))
stack = stack[:-1]
docImage = numpy.hstack(stack).T * 255.0
img = cvarray.array2cv(docImage)
cv.SaveImage('test_lines/docs.png',img)



# Train...
params = dhdp.Params()
params.runs = 1
params.samples = 1
#params.burnIn = 10000
#c.setOneCluster(True)
#c.setCalcBeta(True)


print 'Fitting model...'
p = ProgBar()