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")
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")