Display predictions in categorical form

tanvi0909 · tanvi0909 · commit bd0297ce847f · 2024-10-30T00:43:40.000+05:30
diff --git 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}')
+
