NBA-WAR / app.py
pakooo's picture
Update app.py
8d5e756
"""A simple gradio app to predict NBA player performance this season"""
import gradio as gr
import pandas as pd
from prophet import Prophet
from datasets import load_dataset
pd.options.plotting.backend = "plotly"
# initialize empty players
players = [""]
# initialize empty seasons
seasons = [1977, 2021]
# load data
nba_dataset = load_dataset("andrewkroening/538-NBA-Historical-Raptor")
# initialize a dataframe from the nba dataset
nba_df = pd.DataFrame(nba_dataset["train"])
# make a player_df with seasons and every player for that season
player_df = nba_df[["season", "player_name"]].copy()
def get_players(this_season, df=player_df):
"""Get the players for a given season"""
# get the players for the given season
season_players = df[df["season"] == this_season]["player_name"].unique().tolist()
return gr.Dropdown.update(choices=season_players), gr.update(visible=False)
def get_forecast(this_year, this_player):
"""Get the forecast for a given player and year and the performance for entire career"""
# load data
nba_data_fore = load_dataset("andrewkroening/538-NBA-Historical-Raptor")
# initialize a dataframe from the nba dataset
df = pd.DataFrame(nba_data_fore["train"])
# truncate to the player selected
dataset = df[df["player_name"] == this_player]
player_data = dataset[["season", "war_total"]].copy()
# player_perform = player_df.copy()
# make a list of the seasons the player played in
player_seasons = player_data["season"].unique().tolist()
# make two dfs, one for actual performance and one for the model
# player_perform = player_perform[player_perform["season"] <= year + 5]
player_data = player_data[player_data["season"] <= this_year]
# convert the season column to a datetime object
player_data["season"] = pd.to_datetime(player_data["season"], format="%Y")
# set the df for prophet
player_data.columns = ["ds", "y"]
player_data = player_data.sort_values("ds")
# build the prophet model
m = Prophet(seasonality_mode="multiplicative").fit(player_data)
future = m.make_future_dataframe(periods=5, freq="Y")
forecast = m.predict(future)
# plot the forecast
fig1 = m.plot(forecast, xlabel="Year", ylabel="Wins Above Replacement")
# plot the actual performance
# fig2 = player_perform.plot(
# x="season", y="war_total", title="Actual Performance", template="plotly_white")
# return the figure
return fig1, gr.Dropdown.update(choices=player_seasons), gr.update(visible=True)
with gr.Blocks() as demo:
gr.Markdown(
"""
### 0xpako's NBA Player Performance Predictor.
***It is designed to show a projection for performance (Wins Above Replacement) and compare it to the actual performance over a career.***
***If the projection hangs, it is because the model is taking a long time to run. Refresh the page and give it another shot... get it?***
"""
)
with gr.Row():
year = gr.Slider(1977, 2021, label="Season", interactive=True, step=1)
player = gr.Dropdown(players, label="Player", interactive=True)
with gr.Column(visible=False) as output_col:
gr.Markdown(
"**Below is the player forecast for the selected season plus 5 years. Next to the graph is a dropdown you can use to change the season and update the chart and see how a player's projection has changed over time.**"
)
with gr.Row():
plt = gr.Plot()
season = gr.Dropdown(seasons, label="Season", interactive=True, step=1)
year.change(get_players, inputs=year, outputs=[player, output_col])
player.change(
get_forecast, inputs=[year, player], outputs=[plt, season, output_col]
)
season.change(
get_forecast, inputs=[season, player], outputs=[plt, season, output_col]
)
demo.launch()