James McCool commited on
Commit
2457f50
·
1 Parent(s): 43474bd

Update correlation functions to use dynamic season parameter and adjust calls in correlation matrix display logic.

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +7 -7
src/streamlit_app.py CHANGED
@@ -236,7 +236,7 @@ def seasonlong_build(data_sample):
236
  @st.cache_data(show_spinner=False)
237
  def run_fantasy_corr(data_sample, season):
238
  cor_testing = data_sample
239
- # cor_testing = cor_testing[cor_testing['Season'] == '22024']
240
  date_list = cor_testing['Date'].unique().tolist()
241
  player_list = cor_testing['Player'].unique().tolist()
242
  corr_frame = pd.DataFrame()
@@ -251,9 +251,9 @@ def run_fantasy_corr(data_sample, season):
251
  return corrM
252
 
253
  @st.cache_data(show_spinner=False)
254
- def run_min_corr(data_sample):
255
  cor_testing = data_sample
256
- # cor_testing = cor_testing[cor_testing['Season'] == '22024']
257
  date_list = cor_testing['Date'].unique().tolist()
258
  player_list = cor_testing['Player'].unique().tolist()
259
  corr_frame = pd.DataFrame()
@@ -486,9 +486,9 @@ if selected_tab == 'Correlation Matrix':
486
  working_data = working_data[working_data['spread'] <= spread_var1_t2[1]]
487
  working_data = working_data[working_data['Team'].isin(corr_var1_t2)]
488
  if corr_var == 'Fantasy':
489
- corr_display = run_fantasy_corr(working_data)
490
  elif corr_var == 'Minutes':
491
- corr_display = run_min_corr(working_data)
492
  display.dataframe(corr_display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), height=1000, use_container_width = True)
493
 
494
  elif split_var1_t2 == 'Specific Players':
@@ -502,9 +502,9 @@ if selected_tab == 'Correlation Matrix':
502
  working_data = working_data[working_data['spread'] <= spread_var1_t2[1]]
503
  working_data = working_data[working_data['Player'].isin(corr_var1_t2)]
504
  if corr_var == 'Fantasy':
505
- corr_display = run_fantasy_corr(working_data)
506
  elif corr_var == 'Minutes':
507
- corr_display = run_min_corr(working_data)
508
  display.dataframe(corr_display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
509
  st.download_button(
510
  label="Export Correlations Model",
 
236
  @st.cache_data(show_spinner=False)
237
  def run_fantasy_corr(data_sample, season):
238
  cor_testing = data_sample
239
+ cor_testing = cor_testing[cor_testing['Season'] == season]
240
  date_list = cor_testing['Date'].unique().tolist()
241
  player_list = cor_testing['Player'].unique().tolist()
242
  corr_frame = pd.DataFrame()
 
251
  return corrM
252
 
253
  @st.cache_data(show_spinner=False)
254
+ def run_min_corr(data_sample, season):
255
  cor_testing = data_sample
256
+ cor_testing = cor_testing[cor_testing['Season'] == season]
257
  date_list = cor_testing['Date'].unique().tolist()
258
  player_list = cor_testing['Player'].unique().tolist()
259
  corr_frame = pd.DataFrame()
 
486
  working_data = working_data[working_data['spread'] <= spread_var1_t2[1]]
487
  working_data = working_data[working_data['Team'].isin(corr_var1_t2)]
488
  if corr_var == 'Fantasy':
489
+ corr_display = run_fantasy_corr(working_data, '22025')
490
  elif corr_var == 'Minutes':
491
+ corr_display = run_min_corr(working_data, '22025')
492
  display.dataframe(corr_display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), height=1000, use_container_width = True)
493
 
494
  elif split_var1_t2 == 'Specific Players':
 
502
  working_data = working_data[working_data['spread'] <= spread_var1_t2[1]]
503
  working_data = working_data[working_data['Player'].isin(corr_var1_t2)]
504
  if corr_var == 'Fantasy':
505
+ corr_display = run_fantasy_corr(working_data, '22025')
506
  elif corr_var == 'Minutes':
507
+ corr_display = run_min_corr(working_data, '22025')
508
  display.dataframe(corr_display.style.background_gradient(axis=0).background_gradient(cmap='RdYlGn').format(precision=2), use_container_width = True)
509
  st.download_button(
510
  label="Export Correlations Model",