if trainOrTest == 'train':
myBoost = AdaBoost(300,verbose = False)
TrainX,TrainY,TrainXID = myBoost.getDataFromFile(train_test_file)
myBoost.train(TrainX,TrainY)
pk.dump(myBoost,open(model_file,'wb'))
if trainOrTest == 'test':
try:
myBoost = pk.load(open(model_file,'rb'))
except:
print("output file has not been generated")
if myBoost.isTrained:
Xtest,yTest,XtestID = myBoost.getDataFromFile(train_test_file)
finalPredictions = myBoost.predict(Xtest)
myBoost.writeToFile(XtestID,finalPredictions,'output.txt')
print("Accuracy is: " ,sum(finalPredictions==yTest)/len(yTest))
else:
print("Untrained model being tested")
#train train-data.txt knn_model.txt knn
#test test-data.txt knn_model.txt knn
if model == 'knn' :
if trainOrTest == 'train':
knn.train(train_test_file,model_file)
if trainOrTest == 'test':
try:
myKnn = open(model_file,'rb')
except:
