mherlie commited on
Commit
71d479a
·
1 Parent(s): 3bf5901

modified code for species column

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ with tabs[2]: # Data Visualization
42
 
43
  st.write("### Feature Correlation")
44
  fig, ax = plt.subplots()
45
- sns.heatmap(df.corr(), annot=True, cmap='coolwarm', ax=ax)
46
  st.pyplot(fig)
47
 
48
  with tabs[3]: # Prediction
 
42
 
43
  st.write("### Feature Correlation")
44
  fig, ax = plt.subplots()
45
+ sns.heatmap(df.drop(columns=['species']).corr(), annot=True, cmap='coolwarm', ax=ax)
46
  st.pyplot(fig)
47
 
48
  with tabs[3]: # Prediction