def predict():
print("Enter Team A Id")
teamId = input()
while (len(teamsDF[teamsDF["Team_Id"] == int(teamId)]) == 0):
print("Please Enter valid Team Id")
teamId = input()
print("Enter Team B Id")
opponentId = input()
while (len(teamsDF[teamsDF["Team_Id"] == int(opponentId)]) == 0):
print("Please Enter valid Team Id")
opponentId = input()
print("Which team won the toss?Enter Id")
tossWon = input()
while ((tossWon != teamId) & (tossWon != opponentId)):
print("Please Enter valid Team Id. %s or %s" % (teamId, opponentId))
tossWon = input()
print("Which team bat first?Enter Id")
batFirst = input()
while ((batFirst != teamId) & (batFirst != opponentId)):
print("Please Enter valid Team Id. %s or %s" % (teamId, opponentId))
batFirst = input()
px = d.generatePredictData(int(teamId), int(opponentId), int(tossWon),
int(batFirst))
px = px.drop(['Match_Id'], 1)
px = px.drop(['Match_Won'], 1)
px = preprocessing.scale(px)
lin_svm = svm.LinearSVC()
lin_svm.fit(X_train, y_train)
pred = lin_svm.predict(px)
if (pred[0] == 1):
print("Team A Wins")
else:
print("Team B Wins")
def predict(aa, bb, cc, dd):
teamsDF = pd.read_csv('../data/Team.csv')
matchesTeamsDF = pd.read_csv("../data/gen/match_team.csv")
data = matchesTeamsDF.drop(['Match_Id'], 1)
data = data.drop(['Unnamed: 0'], 1)
X = data.drop(['Match_Won'], 1)
Y = data['Match_Won']
# print(X)
# print(Y)
X = preprocessing.scale(X)
#print(X)
X_train, X_test, y_train, y_test = model_selection.train_test_split(X, Y, test_size=0.25)
print(aa, bb, cc, dd)
# print("Enter Team A Id")
'''teamId = aa
while(len(teamsDF[teamsDF["Team_Id"] == int(teamId)]) == 0):
print("Please Enter valid Team Id")
teamId = aa
#print("Enter Team B Id")
opponentId = bb
while(len(teamsDF[teamsDF["Team_Id"] == int(opponentId)]) == 0):
print("Please Enter valid Team Id")
opponentId = bb
#print("Which team won the toss?Enter Id")
tossWon = cc
while((tossWon != teamId) & (tossWon != opponentId)):
print("Please Enter valid Team Id. %s or %s" % (teamId, opponentId))
tossWon = cc
#print("Which team bat first?Enter Id")
batFirst = dd
while((batFirst != teamId) & (batFirst != opponentId)):
print("Please Enter valid Team Id. %s or %s" % (teamId, opponentId))
batFirst = dd'''
teamId = aa
opponentId = bb
tossWon = cc
batFirst = dd
px = d.generatePredictData(int(teamId),int(opponentId), int(tossWon), int(batFirst))
#print(px)
px = px.drop(['Match_Id'], 1)
px = px.drop(['Match_Won'], 1)
#print(px)
px = preprocessing.scale(px)
#print(px)
lin_svm = svm.LinearSVC()
lin_svm=lin_svm.fit(X_train, y_train)
#print(lin_svm)
pred = lin_svm.predict(px)
print(pred[0])
print(pred)
x=pred
# return A, B
if (x == 1):
print("A")
return "A wins"
else:
print("B")
return "B wins"
def predict(aa, bb, cc, dd):
print(aa, bb, cc, dd)
#print("Enter Team A Id")
'''teamId = aa
while(len(teamsDF[teamsDF["Team_Id"] == int(teamId)]) == 0):
print("Please Enter valid Team Id")
teamId = aa
#print("Enter Team B Id")
opponentId = bb
while(len(teamsDF[teamsDF["Team_Id"] == int(opponentId)]) == 0):
print("Please Enter valid Team Id")
opponentId = bb
#print("Which team won the toss?Enter Id")
tossWon = cc
while((tossWon != teamId) & (tossWon != opponentId)):
print("Please Enter valid Team Id. %s or %s" % (teamId, opponentId))
tossWon = cc
#print("Which team bat first?Enter Id")
batFirst = dd
while((batFirst != teamId) & (batFirst != opponentId)):
print("Please Enter valid Team Id. %s or %s" % (teamId, opponentId))
batFirst = dd'''
px = d.generatePredictData(int(aa), int(bb), int(cc), int(dd))
# print(px)
px = px.drop(['Match_Id'], 1)
px = px.drop(['Match_Won'], 1)
# print(px)
px = preprocessing.scale(px)
# print(px)
lin_svm = svm.LinearSVC()
lin_svm.fit(X_train, y_train)
pred = lin_svm.predict(px)
# return A, B
if (pred[0] == 1):
#print("A")
return "A wins"
else:
#print("B")
return "B wins"
def predict(aa, bb, cc, dd):
import pandas as pd
import numpy as np
import sklearn
from sklearn import preprocessing, model_selection, svm, neighbors
from sklearn.linear_model import LinearRegression
from sklearn.ensemble import RandomForestClassifier
from sklearn.naive_bayes import GaussianNB
import data as d
teamsDF = pd.read_csv('../data/Team.csv')
matchesTeamsDF = pd.read_csv("../data/gen/match_team.csv")
data = matchesTeamsDF.drop(['Match_Id'], 1)
data = data.drop(['Unnamed: 0'], 1)
X = data.drop(['Match_Won'], 1)
Y = data['Match_Won']
# print(X)
# print(Y)
X = preprocessing.scale(X)
X_train, X_test, y_train, y_test = model_selection.train_test_split(
X, Y, test_size=0.25)
print(aa, bb, cc, dd)
# print("Enter Team A Id")
'''teamId = aa
while(len(teamsDF[teamsDF["Team_Id"] == int(teamId)]) == 0):
print("Please Enter valid Team Id")
teamId = aa
#print("Enter Team B Id")
opponentId = bb
while(len(teamsDF[teamsDF["Team_Id"] == int(opponentId)]) == 0):
print("Please Enter valid Team Id")
opponentId = bb
#print("Which team won the toss?Enter Id")
tossWon = cc
while((tossWon != teamId) & (tossWon != opponentId)):
print("Please Enter valid Team Id. %s or %s" % (teamId, opponentId))
tossWon = cc
#print("Which team bat first?Enter Id")
batFirst = dd
while((batFirst != teamId) & (batFirst != opponentId)):
print("Please Enter valid Team Id. %s or %s" % (teamId, opponentId))
batFirst = dd'''
px = d.generatePredictData(int(aa), int(bb), int(cc), int(dd))
print(px)
px = px.drop(['Match_Id'], 1)
px = px.drop(['Match_Won'], 1)
print(px)
px = preprocessing.scale(px)
print(px)
'''svc_svm = svm.SVC()
svc_svm.fit(X_train,y_train)
pred=svc_svm.predict(px)
print(pred)'''
lin_svm = svm.LinearSVC()
lin_svm.fit(X_train, y_train)
pred = lin_svm.predict(px)
print(pred)
# return A, B
x = pred
if (x == 1):
print("A")
return "A"
else:
print("B")
return "B"