From bd0297ce847ffc0ee7b087013a0283e10377ed69 Mon Sep 17 00:00:00 2001
From: Tanvi <168055039+tanvi0909@users.noreply.github.com>
Date: Wed, 30 Oct 2024 00:43:40 +0530
Subject: [PATCH] Display predictions in categorical form

---
 Job Satisfaction Analysis/app.py | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/Job Satisfaction Analysis/app.py b/Job Satisfaction Analysis/app.py
index 19fbd1a..710318c 100644
--- a/Job Satisfaction Analysis/app.py	
+++ b/Job Satisfaction Analysis/app.py	
@@ -80,7 +80,22 @@
     input_df = input_df[features]
 
     # Make prediction
-    prediction = model.predict(input_df)
+    prediction_numeric = model.predict(input_df)[0]
 
     # Display the prediction
-    st.write(f'Predicted Job Satisfaction: {prediction[0]}')
+    satisfaction_categories = [
+    'Extremely dissatisfied', 
+    'Moderately dissatisfied', 
+    'Slightly dissatisfied', 
+    'Neither satisfied nor dissatisfied', 
+    'Slightly satisfied', 
+    'Moderately satisfied', 
+    'Extremely satisfied'
+]
+
+    # Map numeric prediction to category label
+    prediction_label = satisfaction_categories[int(prediction_numeric)]
+
+    # Display the prediction in categorical form
+    st.write(f'Predicted Job Satisfaction: {prediction_label}')
+