xValentim commited on
Commit
fb89f24
·
1 Parent(s): 0b07ad9
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -87,7 +87,7 @@ def model_risco(base_final, X_test, rf_model_indenizacao, rf_model_tempo):
87
  logistic_model = LogisticRegression(max_iter=500)
88
  logistic_model.fit(X_test, (base_final['resultado'] == 'Procedente').astype(int))
89
  y_pred_proba = logistic_model.predict_proba(X_test)[:, 1]
90
- y_pred_indenizacao = rf_model_indenizacao.predict(X_test)
91
  provisionamento_esperado = (y_pred_indenizacao * y_pred_proba).sum()
92
 
93
  # Classificação de Risco
 
87
  logistic_model = LogisticRegression(max_iter=500)
88
  logistic_model.fit(X_test, (base_final['resultado'] == 'Procedente').astype(int))
89
  y_pred_proba = logistic_model.predict_proba(X_test)[:, 1]
90
+ y_pred_indenizacao = logistic_model.predict(X_test)
91
  provisionamento_esperado = (y_pred_indenizacao * y_pred_proba).sum()
92
 
93
  # Classificação de Risco