Update pages/3_EDA_and_Feature_Engineering.py

pages/3_EDA_and_Feature_Engineering.py

@@ -332,9 +332,11 @@ if 'df' in st.session_state:
     
     import matplotlib.pyplot as plt
     # correlation matrix and heatmap
-    corr = df.corr()
+    # corr = df.corr()
+    corr = df[numeric_columns].corr()
     plt.figure(figsize=(20, 10))
     sns.heatmap(corr, annot=True)
+    
     plt.title('Correlation Matrix')
     plt.show()