Update pages/3_EDA_and_Feature_Engineering.py

pages/3_EDA_and_Feature_Engineering.py

@@ -87,9 +87,17 @@ if 'df' in st.session_state:
     st.write("Polynomial Features Created:")
     st.dataframe(poly_df.head())
 
+    # Correlation Heatmap
     # Correlation Heatmap
     st.write("### Correlation Matrix")
-    corr = df.corr()
+    
+    # Select only numerical columns
+    numerical_df = df.select_dtypes(include=['int64', 'float64'])
+    
+    # Compute the correlation matrix
+    corr = numerical_df.corr()
+    
+    # Plot the heatmap
     fig, ax = plt.subplots(figsize=(12, 8))
     sns.heatmap(corr, annot=True, cmap='coolwarm', ax=ax)
     ax.set_title("Correlation Matrix")