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