feature importances

MarkoBrie · Mar 12, 2024 · 1eeda4d · 1eeda4d
1 parent 8f5d228
commit 1eeda4d
Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/3_Streamlit_dashboard/3_STREAMlit_dashboard.py b/3_Streamlit_dashboard/3_STREAMlit_dashboard.py
@@ -241,7 +241,7 @@ def main():
 
         pred = request_prediction(MLFLOW_URI, data)#[0] * 100000
         score = 0 if pred["prediction"] < 0.2 else 1
-        #st.write(pred)
+
         st.write(pred["prediction"], " -> score ", score)
 
         st.write(

diff --git a/main.py b/main.py
@@ -58,7 +58,7 @@ def predict_credit_score(data: DataPoint):
 
         return {
             'prediction': prediction,
-            'importances': importances
+            'probability': 0.8
         }
     except Exception as e:
         error_msg = f"An error occurred during prediction: {str(e)}"