joaomaia commited on
Commit
445da3d
·
verified ·
1 Parent(s): 1e1cef3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -26,19 +26,21 @@ def predict_tiebreak(df):
26
 
27
  # Aplicar o OptimalBinning e transformar os dados
28
  bin_columns = ['Mean_Log_Odds', 'Sum_Prob']
 
 
29
  for column in bin_columns:
30
  optb = joblib.load(f'{column}_binning_tiebreak_model_v1b.pkl')
31
  df[f"{column}_bin"] = optb.transform(df[column], metric="bins")
32
 
33
- # Aplicar o encoder nas variáveis binadas
34
- df[bin_columns] = encoder.transform(df[[f"{col}_bin" for col in bin_columns]])
35
 
36
  # Selecionar as features para predição
37
  features = df[['Mean_Log_Odds_bin', 'Sum_Prob_bin', 'Ratio_Log_Odds']]
38
  df['Probability'] = model.predict_proba(features)[:, 1]
39
 
40
  # Usar o threshold definido
41
- best_threshold = 0.9400000000000005
42
  df['entrada'] = df['Probability'] >= best_threshold
43
  df = df[df['entrada'] == True]
44
 
@@ -66,4 +68,4 @@ gr.Interface(
66
  outputs=outputs,
67
  title="Previsão de Tiebreaks",
68
  description="Faça o upload de um arquivo Excel contendo dados no formato final_df para prever a probabilidade de menos de 1.5 tiebreaks e verificar se deve entrar na aposta."
69
- ).launch()
 
26
 
27
  # Aplicar o OptimalBinning e transformar os dados
28
  bin_columns = ['Mean_Log_Odds', 'Sum_Prob']
29
+ bin_transformed_columns = [f"{col}_bin" for col in bin_columns]
30
+
31
  for column in bin_columns:
32
  optb = joblib.load(f'{column}_binning_tiebreak_model_v1b.pkl')
33
  df[f"{column}_bin"] = optb.transform(df[column], metric="bins")
34
 
35
+ # Aplicar o encoder nas variáveis binadas, na ordem correta
36
+ df[bin_transformed_columns] = encoder.transform(df[bin_transformed_columns])
37
 
38
  # Selecionar as features para predição
39
  features = df[['Mean_Log_Odds_bin', 'Sum_Prob_bin', 'Ratio_Log_Odds']]
40
  df['Probability'] = model.predict_proba(features)[:, 1]
41
 
42
  # Usar o threshold definido
43
+ best_threshold = 0.9420000000000005
44
  df['entrada'] = df['Probability'] >= best_threshold
45
  df = df[df['entrada'] == True]
46
 
 
68
  outputs=outputs,
69
  title="Previsão de Tiebreaks",
70
  description="Faça o upload de um arquivo Excel contendo dados no formato final_df para prever a probabilidade de menos de 1.5 tiebreaks e verificar se deve entrar na aposta."
71
+ ).launch()