Skip to content

Instantly share code, notes, and snippets.

@juliobguedes

juliobguedes/sns_corrplot.py

Created April 6, 2021 18:30
Show Gist options
  • Save juliobguedes/8566fb162e243136142dd67559f58c30 to your computer and use it in GitHub Desktop.
Save juliobguedes/8566fb162e243136142dd67559f58c30 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
sns_corrplot.py
corr = train.corr(method='spearman')
mask = np.zeros_like(corr)
mask[np.triu_indices_from(mask)] = True
fig, ax = plt.subplots()
ax.figure.set_size_inches(10, 10)
sns.heatmap(corr, ax=ax, annot=True, mask=mask, fmt='.2f', cbar=False, square=True, cmap=plt.cm.Blues, center=0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment