Advertisement
Guest User

Untitled

a guest
Jul 5th, 2012
40
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 3.11 KB | None | 0 0
  1.     # read the inital kernel matrix
  2.  
  3.     kernelLinesTR = readMatrix( baseKernelNameTR )
  4.     #kernelLinesTR = np.loadtxt( baseKernelNameTR )
  5.     print kernelLinesTR.shape
  6.  
  7.     #kernelLinesTR0 = readMatrix( os.path.join(kernelPath, 'trainval.jpg.2SsalTotal0.sift.cb.all.trainval.jpg.2SsalTotal
  8. 0.sift.cb.all.histInt') )
  9.    #print kernelLinesTR0.shape
  10.  
  11.    #kernelLinesTR1 = readMatrix(os.path.join(kernelPath, 'trainval.jpg.2SsalTotal1.sift.cb.all.trainval.jpg.2SsalTotal1
  12. .sift.cb.all.histInt') )
  13.    #print kernelLinesTR1.shape
  14.  
  15.    kernelLinesTE= readMatrix(baseKernelNameTE).T
  16.    print kernelLinesTE.shape
  17.  
  18.    #kernelLinesTE0 = readMatrix(os.path.join(kernelPath, 'trainval.jpg.2SsalTotal0.sift.cb.all.test.jpg.2SsalTotal0.sif
  19. t.cb.all.histInt') )
  20.    #print kernelLinesTE0.shape
  21.  
  22.    #kernelLinesTE1 = readMatrix(os.path.join(kernelPath, 'trainval.jpg.2SsalTotal1.sift.cb.all.test.jpg.2SsalTotal1.sif
  23. t.cb.all.histInt') )
  24.    #print kernelLinesTE1.shape
  25.  
  26.    kernelTR = CombinedKernel()
  27.    kernelTR.append_kernel(CustomKernel(kernelLinesTR))
  28.    #kernelTR.append_kernel(CustomKernel(kernelLinesTR0))
  29.    #kernelTR.append_kernel(CustomKernel(kernelLinesTR1))
  30.  
  31.    # create combined test features
  32.    kernelTE = CombinedKernel()
  33.    kernelTE.append_kernel(CustomKernel(kernelLinesTE))
  34.    #kernelTE.append_kernel(CustomKernel(kernelLinesTE0))
  35.    #kernelTE.append_kernel(CustomKernel(kernelLinesTE1))
  36.  
  37.  
  38.    # get the classes
  39.    indexFileExt = '*_' + featFileTR
  40.    indexFiles = glob.glob( os.path.join(indexFilePath, indexFileExt) )
  41.    print indexFilePath, indexFileExt
  42.    print indexFiles
  43.    for f in indexFiles:
  44.      print 'indexFileTR:', f
  45.      for i in range(5):
  46.        print i
  47.        fTE = f.replace(featFileTR, featFileTE)
  48.        #print 'indexFileTE:', fTE
  49.  
  50.        # train file
  51.        indexFile = open(f, 'r')
  52.        indexFilelinesTR = indexFile.read().splitlines()
  53.        indexFile.close()
  54.        # create labels,
  55.        labelsTR = np.array([ float(d.split()[1]) for d in indexFilelinesTR])
  56.        labelsTR[labelsTR==0] = -1
  57.        labelsTR = BinaryLabels(labelsTR)
  58.  
  59.        # test file
  60.        indexFile = open(fTE, 'r')
  61.        indexFilelinesTE = indexFile.read().splitlines()
  62.        indexFile.close()
  63.        # create labels,
  64.        labelsTE = np.array([ float(d.split()[1]) for d in indexFilelinesTE])
  65.        labelsTE[labelsTE==0] = -1
  66.        labelsTE = BinaryLabels(labelsTE)
  67.  
  68.        # train mkl
  69.        mkl = MKLClassification()
  70.  
  71.        mkl.io.set_loglevel(MSG_DEBUG)
  72.        # which norm to use for MKL
  73.        mkl.set_mkl_norm(1) #1, 2,3
  74.  
  75.        # set cost (neg, pos)
  76.        mkl.set_C(1, 1)
  77.        mkl.set_epsilon(1e-5)
  78.  
  79.        # set kernel and labels
  80.        mkl.set_kernel(kernelTR)
  81.        mkl.set_labels(labelsTR)
  82.  
  83.        # train
  84.        mkl.train()
  85.        w=kernelTR.get_subkernel_weights()
  86.        print 'kernel Weights',  w
  87.        # test
  88.        # and classify
  89.        mkl.set_kernel(kernelTE)
  90.        print 'apply'
  91.        mkl.apply()
  92.        print 'getLabels'
  93.        predLabels = mkl.apply().get_labels()
  94.        print predLabels
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement