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: