Update app.py
Browse files
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[
|
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.
|
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()
|