File size: 4,095 Bytes
bb66d8f 9e078f3 bb66d8f c8f4e87 bb66d8f c8f4e87 bb66d8f c8f4e87 bb66d8f 37ba681 bb66d8f a460b33 91fd571 bb66d8f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | import gradio as gr
from joblib import load
import pandas as pd
path='./huggingface_final_df_deployment.csv'
df=pd.read_csv(path)
df['DATE']=pd.to_datetime(df['DATE'])
y_values = ['L', 'W']
y=pd.Series(y_values)
y=y.astype('category')
model=load('./model_LogisticRegression.joblib')
#aquí indicamos cuál es el valor de las clases
label_encoding=load('./label_classes.joblib') # aquí traemos el orden de clasificación de las categoría
idx_labels=label_encoding.transform(y.cat.categories) # reordeno mis categorías con esos índices
class_names=y.cat.categories[idx_labels] # aquí damos los valores de las categorías en el orden que teníamos guardados
def run_my_model(Home_Team, Away_Team): #aquí ponemos que es solo una entrada pero es UNA TABLA la entrada
#Paso1: Función de HomeTeam
def home_team_recent_record(Home_Team):
filtered_df=df[df['TEAM_ABBREVIATION']==Home_Team]
most_recent_record=filtered_df[filtered_df['DATE']==filtered_df['DATE'].max()]
return most_recent_record
#Paso2: Función de AwayTeam
def away_team_recent_record(Away_Team):
filtered_df=df[df['TEAM_ABBREVIATION']==Away_Team]
most_recent_record=filtered_df[filtered_df['DATE']==filtered_df['DATE'].max()]
return most_recent_record
#Paso3: Dataframe con últimos records de HomeTeam
df_home_team_latest_record=home_team_recent_record(Home_Team)
df_home_team_latest_record=df_home_team_latest_record.reset_index()
#Paso4: Dataframe con últimos records de AwayTeam
df_away_team_latest_record=away_team_recent_record(Away_Team)
df_away_team_latest_record=df_away_team_latest_record.reset_index()
#Paso5: Cálculos
HOME_AWAY="HOME"
WINS_DIFFERENTIAL=df_home_team_latest_record.loc[0,"W"]-df_away_team_latest_record.loc[0,"W"]
WINNING_PERCENTAGE_DIFFERENTIAL=(df_home_team_latest_record.loc[0,'W_PCT']-df_away_team_latest_record.loc[0,'W_PCT'])*100
AWAY_WINS_DIFFERENTIAL=df_home_team_latest_record.loc[0,'AWAY_WINS']-df_away_team_latest_record.loc[0,'AWAY_WINS']
WINS_IN_LAST_10_GAMES=df_home_team_latest_record.loc[0,'LAST_10_GAME_WINS']
WINS_IN_LAST_5_GAMES=df_home_team_latest_record.loc[0,'LAST_5_GAME_WINS']
BACK_TO_BACK="NO_B2B"
#Paso6: Convierto a un input dataframe
data = {
'HOME/AWAY': [HOME_AWAY],
'WINS_DIFFERENTIAL': [WINS_DIFFERENTIAL],
'WINNING PERCENTAGE DIFFERENTIAL': [WINNING_PERCENTAGE_DIFFERENTIAL],
'AWAY WINS DIFFERENTIAL': [AWAY_WINS_DIFFERENTIAL],
'WINS IN LAST 10 GAMES': [WINS_IN_LAST_10_GAMES],
'WINS IN LAST 5 GAMES': [WINS_IN_LAST_5_GAMES],
'BACK-TO-BACK': [BACK_TO_BACK]}
input_df = pd.DataFrame(data)
#Paso7: Creación del If
if class_names[model.predict(input_df)][0]=="W":
Winning_Team=Home_Team
Winning_Proba=model.predict_proba(input_df)[0][1]
else:
Winning_Team=Away_Team
Winning_Proba=model.predict_proba(input_df)[0][0]
return f"The team with higher probability of winning is: {Winning_Team} with a probability of {Winning_Proba * 100:.2f}%"
gui= gr.Interface(
fn=run_my_model,
inputs=[gr.Dropdown(['ATL', 'BKN', 'BOS', 'CHA', 'CHI', 'CLE', 'DAL', 'DEN', 'DET', 'GSW', 'HOU', 'IND', 'LAC', 'LAL', 'MEM', 'MIA', 'MIL', 'MIN', 'NOP', 'NYK', 'OKC', 'ORL', 'PHI', 'PHX', 'POR', 'SAC', 'SAS', 'TOR', 'UTA', 'WAS'],label="HOME TEAM"),gr.Dropdown(['ATL', 'BKN', 'BOS', 'CHA', 'CHI', 'CLE', 'DAL', 'DEN', 'DET', 'GSW', 'HOU', 'IND', 'LAC', 'LAL', 'MEM', 'MIA', 'MIL', 'MIN', 'NOP', 'NYK', 'OKC', 'ORL', 'PHI', 'PHX', 'POR', 'SAC', 'SAS', 'TOR', 'UTA', 'WAS'],label="AWAY TEAM")],
outputs="text",
live=False,
title='NBA regular season games outcome prediction',
description='IMPORTANT: The predictions generated by this AI tool are for informational purposes only and should not be considered as any type of advice. Past performance is not indicative of future results. This predictions are only for NBA regular season games.'
)
gui.launch() |