Update app.py

adicionando entrada_Home e entrada_Away

app.py

@@ -11,7 +11,9 @@ def fazer_previsao(ODD_H, ODD_D, ODD_A):
     data = pd.DataFrame({'H/A': [ODD_H / ODD_A], 'Diff': [(ODD_H / ODD_D) - (ODD_H / ODD_A)]})
     previsao_LayAway = model_LayAway.predict(data)[0]
     previsao_LayHome = model_LayHome.predict(data)[0]
-
+    #previsao_H = model_LayAway.predict_proba(data)[1]
+    #previsao_A = model_LayHome.predict_proba(data)[1]
+    
     if (previsao_LayAway == previsao_LayHome):
         resultado = 'Não entre!'
     else:
@@ -28,11 +30,26 @@ def fazer_previsao(ODD_H, ODD_D, ODD_A):
         probabilidade_LayAway = round(100 * model_LayAway.predict_proba(data)[0][1], 2)
         probabilidade_LayHome = round(100 * model_LayHome.predict_proba(data)[0][1], 2)
 
+        if probabilidade_LayAway > 0.682:
+            entrada_Home = 'SIM'
+        else:
+            entrada_Home = 'NAO'
+
+        if (probabilidade_LayAway < 0.261) & (probabilidade_LayHome > 0.885):
+            entrada_Away = 'SIM'
+        else:
+            entrada_Away = 'NAO'
+            
+            
         resultado = f"entrada LayAway (se Odd> 1.17): {entrada_LayAway}\n" \
                     f"------ probabilidade: {probabilidade_LayAway}%\n" \
                     f"\n" \
+                    f"entrada Home: {entrada_Home}\n" \
+                    f"\n" \
                     f"entrada LayHome (se Odd> 1.28): {entrada_LayHome}\n" \
                     f"------ probabilidade: {probabilidade_LayHome}%"
+                    f"\n" \
+                    f"entrada Home: {entrada_Away}\n" \
 
     return resultado