def getResults(ground, innings, venue, team1, team2, choice): ground=ground[1:-1] team1=team1[1:-1] team2=team2[1:-1] SVMClf = ourSVMClassifier() MLPClf = ourMLPClassifier() DTClf = ourDTClassifier() dIF = dataInputFormat.DataInput_Categorical() print("Starting Up..!\n\nLoading Pickled Models") SVMClf.loadPickle() DTClf.loadPickle() MLPClf.loadPickle() dIF.hashAll() inputPrediction = [0]*217 if innings == 1: inningsTeam1 = "Team1_1Inning" inningsTeam2 = "Team2_2Inning" elif innings == 2: inningsTeam1 = "Team1_2Inning" inningsTeam2 = "Team2_1Inning" if venue == 1: venueTeam1 = "Team1_Home" venueTeam2 = "Team2_Away" elif venue == 2: venueTeam1 = "Team1_Away" venueTeam2 = "Team2_Home" elif venue == 3: venueTeam1 = "Team1_Neutral" venueTeam2 = "Team2_Neutral" inputPrediction[dIF.ourTeams_1[team1]] = 1 inputPrediction[dIF.ourTeams_2[team2]] = 1 inputPrediction[dIF.ourGrounds[ground]] = 1 inputPrediction[dIF.ourInnings[inningsTeam1]] = 1 inputPrediction[dIF.ourInnings[inningsTeam2]] = 1 inputPrediction[dIF.ourVenues[venueTeam1]] = 1 inputPrediction[dIF.ourVenues[venueTeam2]] = 1 if choice==1: return SVMClf.runModel(inputPrediction,team1,team2) elif choice==2: return DTClf.runModel(inputPrediction,team1,team2) elif choice==3: return MLPClf.runModel(inputPrediction,team1,team2)
import sys import pickle from Classifier import ourMLPClassifier, ourDTClassifier, ourSVMClassifier if __name__ == '__main__': MLPClf = ourMLPClassifier() DTClf = ourDTClassifier() SVMClf = ourSVMClassifier() print("Beginning to Pickle MLP Model!") MLPClf.trainModel() MLPClf.accuracyCheck() MLPClf.dumpPickle() print("Done: MLP Model!") print("Beginning to Pickle DT Model!") DTClf.trainModel() DTClf.accuracyCheck() DTClf.dumpPickle() print("Done: DT Model!\n") print("Beginning to Pickle SVM Model!") SVMClf.trainModel() SVMClf.accuracyCheck() SVMClf.dumpPickle() print("Done: SVM Model!\n")