app

app.py

@@ -32,8 +32,7 @@ def outbreak(plot_type):
         return fig
     if plot_type == "Age vs Attrition":
         fig = plt.figure()
-        positive_attrition_df = data_encoded.loc[data_encoded['Attrition'] == "Yes"]
-        plt.hist(positive_attrition_df['Age'], bins=np.arange(0,80,10), alpha=0.8, rwidth=0.9, color='blue')
+        plt.hist(data_encoded['Age'], bins=np.arange(0,80,10), alpha=0.8, rwidth=0.9, color='red')
         plt.xlabel("Age")
         plt.ylabel("Count")
         plt.title("Age vs Attrition")