From 19033b4eec19d9507786f455b6f5e8c4189bc39c Mon Sep 17 00:00:00 2001 From: Zhang Zhanpeng Date: Wed, 30 Jun 2021 18:13:37 +0800 Subject: [PATCH] Made the logarithmic image of cross_correlation more readable --- analysis.ipynb | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/analysis.ipynb b/analysis.ipynb index a785b863..ae8b8611 100644 --- a/analysis.ipynb +++ b/analysis.ipynb @@ -693,7 +693,8 @@ "genres_indicator = enc.fit_transform(tracks['track', 'genres'])\n", "genres_names = enc.classes_\n", "genres_names = genres.loc[enc.classes_, 'title'].values\n", - "cross_correlation = genres_indicator.T @ genres_indicator" + "cross_correlation = genres_indicator.T @ genres_indicator\n", + "np.fill_diagonal(cross_correlation, 0)" ] }, { @@ -702,11 +703,9 @@ "metadata": {}, "outputs": [], "source": [ - "np.fill_diagonal(cross_correlation, 0)\n", - "\n", "plt.figure(figsize=(28, 28))\n", - "plt.imshow(np.log(cross_correlation))\n", - "plt.yticks(range(len(genres_names)), genres_names);\n", + "plt.imshow(np.log(cross_correlation[::-1]))\n", + "plt.yticks(range(len(genres_names)), genres_names[::-1]);\n", "plt.xticks(range(len(genres_names)), genres_names, rotation=90);" ] },