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()