import matplotlib.pyplot as plt
import seaborn as sns
figure = plt.figure(figsize=(12, 6))
sns.heatmap(train_data.corr(), annot=True,cmap=plt.cm.cool)
plt.tight_layout()
plt.xlabel('Corr')
plt.show()
import matplotlib.pyplot as plt
import seaborn as sns
figure = plt.figure(figsize=(12, 6))
sns.heatmap(df.corr(), annot=True,cmap=plt.cm.cool)
plt.tight_layout()
plt.xlabel('Corr')
plt.show()