Ejemplo n.º 1
0
def run( inputDataFilepath, inputTrainingFilepath, outputFilepath, outputDirpath, numClusters, numComponents ):

    c_dest_dir  = outputDirpath + "/"
    train 	= pickle.load(open(inputTrainingFilepath))
    test 	= pickle.load(open(inputDataFilepath))

    print "making triple ... \n"
    triple = makeTriple(train, test, numClusters, numComponents)
    print "dumping pickle ... \n"
    pickle.dump(triple, open(c_dest_dir + "triple.pkl", "w"))
    for i in range(0,10):
        confPair = confusion(triple,i)
        print "dumping confusion", str(i)
        pickle.dump( confPair, open( c_dest_dir + "conf_" + str(i) + ".pkl","w" ) )
        #print "writing confusion matrix in text file", str(i)
        #writeConfusion( confPair[1], c_dest_dir + 
        #    "conf_" + str(i) + ".txt" )
        montagelist, confusionarray = confPair
        print "\n Making montage", str(i)
        montage.colorsMontage( montagelist, c_dest_dir + str(i) + ".png")	
Ejemplo n.º 2
0
        montage.colorsMontage( montagelist, c_dest_dir + str(i) + ".png")	

if __name__ == "__main__": 
	c_trainf, c_testf, c_dir = sys.argv[1:4]
	k = int(sys.argv[4])
	D = int(sys.argv[5]) 
	c_basename = os.path.basename(c_testf).split(".")[0]
	c_triple = c_basename + "_triple.pkl"
	c_confusion = c_basename + "_confusion.pkl"
	c_confusiontext = c_basename + "_confusion.txt" 
	train 	= pickle.load(open(c_trainf)) 
	test 	= pickle.load(open(c_testf))   
	#Save this
	print "making triple ... \n" 
	triple = makeTriple(train,test,k,D)
	print "dumping pickle ... \n"
	pickle.dump(triple,open(c_dir + c_triple, "w")) 
	for i in range(0,10):
		confPair = confusion(triple,i)
		print "dumping confusion", str(i)
		pickle.dump( confPair, open( c_dir + c_basename + \
			 "_confusion" + str(i) + ".pkl","w" ) )
		print "writing confusion matrix in text file", str(i)
		writeConfusion( confPair[1], c_dir + c_basename + \
			"_confusion" + str(i) + ".txt" )
		montagelist, confusionarray = confPair
		print "\n Making montage", str(i)
		montage.colorsMontage( montagelist, c_dir + c_basename +\
			"_montage" + str(i) + ".jpg")