Class.test(generator, v=True)
else:
classFile = open("class.pkl","rb")
Class = pickle.load(classFile)
#WRITE SUBMISSION
input("Go on with submission?")
outputFile = open(OUT_PATH, "w")
outputFile.write("id,click\n")
i = 0
gen = testGenerator(SUB_PATH, NUM_TEST_BATCH, featCreators = funcs)
predictionGen = Class.generatePrediction(gen)
for pBatch, idBatch in predictionGen:
outputText = ""
for example in range(len(pBatch)):
ID = idBatch[example]
click = pBatch[example][1] #Class 1
outputText += str(ID) + "," + str(click) + "\n"
outputFile.write(outputText)
print(100 * i / NUM_TEST_BATCH, "%")
i += 1
outputFile.close()
print("done")
