ListeningBrainzColFilt_RecomModel.ipynb

{"cells":[{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"elapsed":20911,"status":"ok","timestamp":1679183642175,"user":{"displayName":"ALBERTO BARRERA HERRERO","userId":"09905877085308943679"},"user_tz":-60},"id":"6tgQIhcHkYMO","outputId":"738e0bf9-77db-4833-952e-54ad5dfc5bf3"},"outputs":[{"output_type":"stream","name":"stdout","text":["Mounted at /content/drive\n"]}],"source":["# I'm working on Google Drive so I mount my Drive into Colab and change the working directory to the project folder.\n","import os\n","from google.colab import drive\n","drive.mount('/content/drive')\n","path = '/content/drive/MyDrive/AMPLab/LargeScaleDatasets/CollaborativeFiltering'\n","os.chdir(path)"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"h6LdkyzXBFhW"},"outputs":[],"source":["from scipy import sparse\n","\n","import implicit\n","import csv\n","import numpy as np\n","import pandas as pd"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"elapsed":12132,"status":"ok","timestamp":1679183621270,"user":{"displayName":"ALBERTO BARRERA HERRERO","userId":"09905877085308943679"},"user_tz":-60},"id":"yRpJBZ0kkOxM","outputId":"0f56ce5f-5b95-4113-da5d-738fb7f68ac9"},"outputs":[{"output_type":"stream","name":"stdout","text":["Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n","Collecting implicit\n","  Downloading implicit-0.6.2-cp39-cp39-manylinux2014_x86_64.whl (18.6 MB)\n","\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m18.6/18.6 MB\u001b[0m \u001b[31m43.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n","\u001b[?25hRequirement already satisfied: tqdm>=4.27 in /usr/local/lib/python3.9/dist-packages (from implicit) (4.65.0)\n","Requirement already satisfied: numpy in /usr/local/lib/python3.9/dist-packages (from implicit) (1.22.4)\n","Requirement already satisfied: scipy>=0.16 in /usr/local/lib/python3.9/dist-packages (from implicit) (1.10.1)\n","Installing collected packages: implicit\n","Successfully installed implicit-0.6.2\n"]}],"source":["# I'll be using the implicit library to manage the sparse matrix.\n","!pip install implicit"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"elapsed":420,"status":"ok","timestamp":1679183644301,"user":{"displayName":"ALBERTO BARRERA HERRERO","userId":"09905877085308943679"},"user_tz":-60},"id":"dwUaycQMG8If","outputId":"ec7b4eb5-e0e0-4363-b23c-a005b3251a34"},"outputs":[{"output_type":"stream","name":"stdout","text":["mbid,name\n","ca3f3ee1-c4a7-4bac-a16a-0b888a396c6b,The Silhouettes\n","b68a3969-319a-462f-942b-cd35581414fc,Evie Tamala\n","4c4b7c6f-9285-4d6a-bc10-e5c9e08045f8,wecamewithbrokenteeth\n","ab1b631b-9896-4433-bef9-7868bf8a42f3,Giant Tomo\n","66de1369-f9eb-43cb-ae4f-88582a47a624,Elvin Jones & Jimmy Garrison Sextet\n","3175ff46-8e09-4562-ace2-0e557e0fff42,Disappointment Incorporated\n","872a1247-c0d8-4970-b608-6752db0e532b,Stereobate\n","1c2d8a03-9a9f-4424-be44-61fd5634ea1d,We Barbarians\n","9047f366-3560-40d5-bbcc-fba04118737d,Nodesha\n"]}],"source":["!head /content/drive/MyDrive/AMPLab/LargeScaleDatasets/CollaborativeFiltering/subset_artistmbid_name.csv"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"SjnCdoFaHSdJ"},"outputs":[],"source":["## We make a dictionary for both converting from artist_name to artist_mbid and vice versa\n","\n","artist_name_2_mbid = {}\n","artist_mbid_2_name = {}\n","\n","with open('subset_artistmbid_name.csv', 'r') as fr:\n","    reader = csv.DictReader(fr)\n","    for line in reader:\n","        if line['mbid'] not in artist_name_2_mbid.keys():\n","            artist_mbid_2_name[line['mbid']] = [line['name']]\n","        else:\n","            artist_mbid_2_name[line['mbid']].append(line['name'])\n","\n","        if line['name'] not in artist_name_2_mbid.keys():\n","            artist_name_2_mbid[line['name']] = [line['mbid']]\n","        else:\n","            artist_name_2_mbid[line['name']].append(line['mbid'])"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":35},"executionInfo":{"elapsed":15,"status":"ok","timestamp":1679183646874,"user":{"displayName":"ALBERTO BARRERA HERRERO","userId":"09905877085308943679"},"user_tz":-60},"id":"6LCaqZwcIXMa","outputId":"0c523a6b-9acb-464c-9bd7-13a6c8725cd8"},"outputs":[{"output_type":"execute_result","data":{"text/plain":["'c3f28da8-662d-4f09-bdc7-3084bf685930'"],"application/vnd.google.colaboratory.intrinsic+json":{"type":"string"}},"metadata":{},"execution_count":6}],"source":["# Check that it works with an artist name.\n","artist_name_2_mbid['Iron & Wine'][0]"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"oOY4kOpwLUcb"},"outputs":[],"source":["# Read the userID, artist_mbid, artist name and playcount file to a Pandas Dataframe.\n","\n","df = pd.read_table(\n","    'userID_artistmbid_artistName_plays.tsv', usecols=[0, 1, 3], skiprows=1, names=['user', 'artist_mbid', 'plays'], na_filter=False\n",")"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"uxCNyTygMSS-"},"outputs":[],"source":["# Data is set as categorical and artist_mbid is factorized so that it can be turnt into the Sparse Matrix.\n","\n","df['user'] = df['user'].astype('category')\n","df['artist_mbid'] = df['artist_mbid'].astype('category')\n","df['plays'] = df['plays'].astype(int)\n","df['artist_index'] = pd.factorize(df['artist_mbid'])[0]"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"KuErE9abGJ_p"},"outputs":[],"source":["# To be able to switch between artist_mbid and artist_index, both an artist_index to artist_mbid and an artist_mbid to artist_index dictionaries are created.\n","\n","artist_index2artist_mbid = df.set_index('artist_index')['artist_mbid'].to_dict()\n","\n","artist_mbid_2_index = {}\n","\n","for key, value in artist_index2artist_mbid.items():\n","    artist_mbid_2_index[value] = key"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"elapsed":9,"status":"ok","timestamp":1679183664940,"user":{"displayName":"ALBERTO BARRERA HERRERO","userId":"09905877085308943679"},"user_tz":-60},"id":"aKGQVQd9DKN7","outputId":"4d0a013c-ec90-48d7-8f60-77725f449cf1"},"outputs":[{"output_type":"execute_result","data":{"text/plain":["['Iron & Wine']"]},"metadata":{},"execution_count":11}],"source":["# We check again that everything is working\n","\n","artist_mbid_2_name[artist_index2artist_mbid[6245]]"]},{"cell_type":"code","source":["artist_mbid_2_index[artist_name_2_mbid['Iron & Wine'][0]]"],"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"id":"SymmXlbW4BhD","executionInfo":{"status":"ok","timestamp":1679183773388,"user_tz":-60,"elapsed":33,"user":{"displayName":"ALBERTO BARRERA HERRERO","userId":"09905877085308943679"}},"outputId":"36865d39-1b7d-4bae-97a0-d52121af8d50"},"execution_count":null,"outputs":[{"output_type":"execute_result","data":{"text/plain":["6245"]},"metadata":{},"execution_count":16}]},{"cell_type":"code","execution_count":null,"metadata":{"id":"engiVenGsv7I"},"outputs":[],"source":["userid_2_index = {user: index for index, user in enumerate(set(df['user']))}\n","user_indexes = np.array([userid_2_index[user] for user in df['user']])\n","artist_indexes = df['artist_index']"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"UJTtKKBGNGwS"},"outputs":[],"source":["# create a sparse matrix of all the users/plays\n","from scipy.sparse import coo_matrix, csr_matrix\n","plays = coo_matrix(\n","    (\n","        df['plays'].astype(np.float32),\n","        (artist_indexes, user_indexes)\n","    )\n",")"]},{"cell_type":"code","execution_count":null,"metadata":{"id":"NjPHU4ZkULAx"},"outputs":[],"source":["from implicit.nearest_neighbours import bm25_weight\n","\n","artist_user_plays = bm25_weight(plays, K1=100, B=0.8)\n","user_plays = artist_user_plays.T.tocsr()"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":49,"referenced_widgets":["c1e146def1d44e40bdfadaa22e4f31aa","d1173548d5db4ece9c6db841cea9f4eb","04741fb123ae499c870c105f5f6d08a8","59a5360c93064f61905f9f8abc0aa3d8","323689fcce7e4dc5b1ec17f638381a37","06a8eda9faa94df9adc394505f18df9e","1120c521aa7c43caa50bf32dc74e1331","61c65e39ed164a579bf33b8fadcb40b5","2a6638270cc44ff2983ce87308e5802d","44e4c4f101de422d9ad53ecc8b3e9c5f","97739611e8f74363b0d929adc5dccaae"]},"id":"GQ9NI1nubAmK","executionInfo":{"status":"ok","timestamp":1679183773387,"user_tz":-60,"elapsed":106390,"user":{"displayName":"ALBERTO BARRERA HERRERO","userId":"09905877085308943679"}},"outputId":"1c819510-4571-42ce-d1e7-7c588f9448d7"},"outputs":[{"output_type":"display_data","data":{"text/plain":["  0%|          | 0/15 [00:00<?, ?it/s]"],"application/vnd.jupyter.widget-view+json":{"version_major":2,"version_minor":0,"model_id":"c1e146def1d44e40bdfadaa22e4f31aa"}},"metadata":{}}],"source":["# Train the model with alternating least squares of the different users.\n","\n","from implicit.als import AlternatingLeastSquares\n","\n","model = AlternatingLeastSquares(factors=1000, regularization=0.05, alpha=2.0)\n","model.fit(user_plays)"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":450},"executionInfo":{"elapsed":12,"status":"ok","timestamp":1679183773389,"user":{"displayName":"ALBERTO BARRERA HERRERO","userId":"09905877085308943679"},"user_tz":-60},"id":"PXWau_Uub4iw","outputId":"ed82768a-8f1f-4622-d5ab-bc4e2f2d86c5"},"outputs":[{"output_type":"stream","name":"stderr","text":["<ipython-input-17-99e55f7f707c>:12: SettingWithCopyWarning: \n","A value is trying to be set on a copy of a slice from a DataFrame\n","\n","See the caveats in the documentation: https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#returning-a-view-versus-a-copy\n","  recom['artist_name'].iloc[i] = artist_mbid_2_name[artist_index2artist_mbid[recom['artist'].iloc[i]]][0]\n"]},{"output_type":"execute_result","data":{"text/plain":["      score             artist_name\n","0  1.000000             Iron & Wine\n","1  0.422283     Gregory Alan Isakov\n","2  0.390881             Fleet Foxes\n","3  0.383695               The Shins\n","4  0.376312                Bon Iver\n","5  0.375864           José González\n","6  0.369226          Sufjan Stevens\n","7  0.365435              Lord Huron\n","8  0.360813     Death Cab for Cutie\n","9  0.341159  The Head and the Heart"],"text/html":["\n","  <div id=\"df-d21408ba-2017-45b5-b1ed-4ae4a41d1875\">\n","    <div class=\"colab-df-container\">\n","      <div>\n","<style scoped>\n","    .dataframe tbody tr th:only-of-type {\n","        vertical-align: middle;\n","    }\n","\n","    .dataframe tbody tr th {\n","        vertical-align: top;\n","    }\n","\n","    .dataframe thead th {\n","        text-align: right;\n","    }\n","</style>\n","<table border=\"1\" class=\"dataframe\">\n","  <thead>\n","    <tr style=\"text-align: right;\">\n","      <th></th>\n","      <th>score</th>\n","      <th>artist_name</th>\n","    </tr>\n","  </thead>\n","  <tbody>\n","    <tr>\n","      <th>0</th>\n","      <td>1.000000</td>\n","      <td>Iron &amp; Wine</td>\n","    </tr>\n","    <tr>\n","      <th>1</th>\n","      <td>0.422283</td>\n","      <td>Gregory Alan Isakov</td>\n","    </tr>\n","    <tr>\n","      <th>2</th>\n","      <td>0.390881</td>\n","      <td>Fleet Foxes</td>\n","    </tr>\n","    <tr>\n","      <th>3</th>\n","      <td>0.383695</td>\n","      <td>The Shins</td>\n","    </tr>\n","    <tr>\n","      <th>4</th>\n","      <td>0.376312</td>\n","      <td>Bon Iver</td>\n","    </tr>\n","    <tr>\n","      <th>5</th>\n","      <td>0.375864</td>\n","      <td>José González</td>\n","    </tr>\n","    <tr>\n","      <th>6</th>\n","      <td>0.369226</td>\n","      <td>Sufjan Stevens</td>\n","    </tr>\n","    <tr>\n","      <th>7</th>\n","      <td>0.365435</td>\n","      <td>Lord Huron</td>\n","    </tr>\n","    <tr>\n","      <th>8</th>\n","      <td>0.360813</td>\n","      <td>Death Cab for Cutie</td>\n","    </tr>\n","    <tr>\n","      <th>9</th>\n","      <td>0.341159</td>\n","      <td>The Head and the Heart</td>\n","    </tr>\n","  </tbody>\n","</table>\n","</div>\n","      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-d21408ba-2017-45b5-b1ed-4ae4a41d1875')\"\n","              title=\"Convert this dataframe to an interactive table.\"\n","              style=\"display:none;\">\n","        \n","  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n","       width=\"24px\">\n","    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n","    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n","  </svg>\n","      </button>\n","      \n","  <style>\n","    .colab-df-container {\n","      display:flex;\n","      flex-wrap:wrap;\n","      gap: 12px;\n","    }\n","\n","    .colab-df-convert {\n","      background-color: #E8F0FE;\n","      border: none;\n","      border-radius: 50%;\n","      cursor: pointer;\n","      display: none;\n","      fill: #1967D2;\n","      height: 32px;\n","      padding: 0 0 0 0;\n","      width: 32px;\n","    }\n","\n","    .colab-df-convert:hover {\n","      background-color: #E2EBFA;\n","      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n","      fill: #174EA6;\n","    }\n","\n","    [theme=dark] .colab-df-convert {\n","      background-color: #3B4455;\n","      fill: #D2E3FC;\n","    }\n","\n","    [theme=dark] .colab-df-convert:hover {\n","      background-color: #434B5C;\n","      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n","      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n","      fill: #FFFFFF;\n","    }\n","  </style>\n","\n","      <script>\n","        const buttonEl =\n","          document.querySelector('#df-d21408ba-2017-45b5-b1ed-4ae4a41d1875 button.colab-df-convert');\n","        buttonEl.style.display =\n","          google.colab.kernel.accessAllowed ? 'block' : 'none';\n","\n","        async function convertToInteractive(key) {\n","          const element = document.querySelector('#df-d21408ba-2017-45b5-b1ed-4ae4a41d1875');\n","          const dataTable =\n","            await google.colab.kernel.invokeFunction('convertToInteractive',\n","                                                     [key], {});\n","          if (!dataTable) return;\n","\n","          const docLinkHtml = 'Like what you see? Visit the ' +\n","            '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n","            + ' to learn more about interactive tables.';\n","          element.innerHTML = '';\n","          dataTable['output_type'] = 'display_data';\n","          await google.colab.output.renderOutput(dataTable, element);\n","          const docLink = document.createElement('div');\n","          docLink.innerHTML = docLinkHtml;\n","          element.appendChild(docLink);\n","        }\n","      </script>\n","    </div>\n","  </div>\n","  "]},"metadata":{},"execution_count":17}],"source":["# Check for a known artist\n","id = artist_mbid_2_index[artist_name_2_mbid['Iron & Wine'][0]]\n","ids, scores= model.similar_items(id)\n","\n","# Display the results using pandas for nicer formatting\n","recom = pd.DataFrame({\"artist\": ids, \"score\": scores})\n","recom['artist_name'] = 0 * recom['artist']\n","\n","for i in range(len(recom['artist'])):\n","    recom['artist_name'].iloc[i] = artist_mbid_2_name[artist_index2artist_mbid[recom['artist'].iloc[i]]][0]\n","\n","recom = recom.drop(['artist'], axis=1)\n","recom"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":450},"id":"4IrPTINsJYZU","executionInfo":{"status":"ok","timestamp":1679183994076,"user_tz":-60,"elapsed":295,"user":{"displayName":"ALBERTO BARRERA HERRERO","userId":"09905877085308943679"}},"outputId":"4e838694-7e80-4241-f377-aa29f5433905"},"outputs":[{"output_type":"stream","name":"stderr","text":["<ipython-input-30-f87372b5e942>:10: SettingWithCopyWarning: \n","A value is trying to be set on a copy of a slice from a DataFrame\n","\n","See the caveats in the documentation: https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#returning-a-view-versus-a-copy\n","  recom['artist_name'].iloc[i] = artist_mbid_2_name[artist_index2artist_mbid[recom['artist'].iloc[i]]][0]\n"]},{"output_type":"execute_result","data":{"text/plain":["      score     artist_name\n","0  1.000000            高中正義\n","1  0.386140        Casiopea\n","2  0.328007            大貫妙子\n","3  0.326699        T‐SQUARE\n","4  0.301526              杏里\n","5  0.278965            Lamp\n","6  0.277665  Naniwa Express\n","7  0.277088           Piper\n","8  0.269268    Serge Ponsar\n","9  0.268987          TUCKER"],"text/html":["\n","  <div id=\"df-0334dbb3-4247-49e2-8dd5-10bd0f40d0fe\">\n","    <div class=\"colab-df-container\">\n","      <div>\n","<style scoped>\n","    .dataframe tbody tr th:only-of-type {\n","        vertical-align: middle;\n","    }\n","\n","    .dataframe tbody tr th {\n","        vertical-align: top;\n","    }\n","\n","    .dataframe thead th {\n","        text-align: right;\n","    }\n","</style>\n","<table border=\"1\" class=\"dataframe\">\n","  <thead>\n","    <tr style=\"text-align: right;\">\n","      <th></th>\n","      <th>score</th>\n","      <th>artist_name</th>\n","    </tr>\n","  </thead>\n","  <tbody>\n","    <tr>\n","      <th>0</th>\n","      <td>1.000000</td>\n","      <td>高中正義</td>\n","    </tr>\n","    <tr>\n","      <th>1</th>\n","      <td>0.386140</td>\n","      <td>Casiopea</td>\n","    </tr>\n","    <tr>\n","      <th>2</th>\n","      <td>0.328007</td>\n","      <td>大貫妙子</td>\n","    </tr>\n","    <tr>\n","      <th>3</th>\n","      <td>0.326699</td>\n","      <td>T‐SQUARE</td>\n","    </tr>\n","    <tr>\n","      <th>4</th>\n","      <td>0.301526</td>\n","      <td>杏里</td>\n","    </tr>\n","    <tr>\n","      <th>5</th>\n","      <td>0.278965</td>\n","      <td>Lamp</td>\n","    </tr>\n","    <tr>\n","      <th>6</th>\n","      <td>0.277665</td>\n","      <td>Naniwa Express</td>\n","    </tr>\n","    <tr>\n","      <th>7</th>\n","      <td>0.277088</td>\n","      <td>Piper</td>\n","    </tr>\n","    <tr>\n","      <th>8</th>\n","      <td>0.269268</td>\n","      <td>Serge Ponsar</td>\n","    </tr>\n","    <tr>\n","      <th>9</th>\n","      <td>0.268987</td>\n","      <td>TUCKER</td>\n","    </tr>\n","  </tbody>\n","</table>\n","</div>\n","      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-0334dbb3-4247-49e2-8dd5-10bd0f40d0fe')\"\n","              title=\"Convert this dataframe to an interactive table.\"\n","              style=\"display:none;\">\n","        \n","  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n","       width=\"24px\">\n","    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n","    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n","  </svg>\n","      </button>\n","      \n","  <style>\n","    .colab-df-container {\n","      display:flex;\n","      flex-wrap:wrap;\n","      gap: 12px;\n","    }\n","\n","    .colab-df-convert {\n","      background-color: #E8F0FE;\n","      border: none;\n","      border-radius: 50%;\n","      cursor: pointer;\n","      display: none;\n","      fill: #1967D2;\n","      height: 32px;\n","      padding: 0 0 0 0;\n","      width: 32px;\n","    }\n","\n","    .colab-df-convert:hover {\n","      background-color: #E2EBFA;\n","      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n","      fill: #174EA6;\n","    }\n","\n","    [theme=dark] .colab-df-convert {\n","      background-color: #3B4455;\n","      fill: #D2E3FC;\n","    }\n","\n","    [theme=dark] .colab-df-convert:hover {\n","      background-color: #434B5C;\n","      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n","      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n","      fill: #FFFFFF;\n","    }\n","  </style>\n","\n","      <script>\n","        const buttonEl =\n","          document.querySelector('#df-0334dbb3-4247-49e2-8dd5-10bd0f40d0fe button.colab-df-convert');\n","        buttonEl.style.display =\n","          google.colab.kernel.accessAllowed ? 'block' : 'none';\n","\n","        async function convertToInteractive(key) {\n","          const element = document.querySelector('#df-0334dbb3-4247-49e2-8dd5-10bd0f40d0fe');\n","          const dataTable =\n","            await google.colab.kernel.invokeFunction('convertToInteractive',\n","                                                     [key], {});\n","          if (!dataTable) return;\n","\n","          const docLinkHtml = 'Like what you see? Visit the ' +\n","            '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n","            + ' to learn more about interactive tables.';\n","          element.innerHTML = '';\n","          dataTable['output_type'] = 'display_data';\n","          await google.colab.output.renderOutput(dataTable, element);\n","          const docLink = document.createElement('div');\n","          docLink.innerHTML = docLinkHtml;\n","          element.appendChild(docLink);\n","        }\n","      </script>\n","    </div>\n","  </div>\n","  "]},"metadata":{},"execution_count":30}],"source":["# Check for Masayoshi Takanaka (高中正義)\n","id = artist_mbid_2_index[artist_name_2_mbid['高中正義'][0]]\n","ids, scores= model.similar_items(id)\n","\n","# Display the results using pandas for nicer formatting\n","recom = pd.DataFrame({\"artist\": ids, \"score\": scores})\n","recom['artist_name'] = 0 * recom['artist']\n","\n","for i in range(len(recom['artist'])):\n","    recom['artist_name'].iloc[i] = artist_mbid_2_name[artist_index2artist_mbid[recom['artist'].iloc[i]]][0]\n","\n","recom = recom.drop(['artist'], axis=1)\n","recom"]},{"cell_type":"code","execution_count":null,"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":450},"executionInfo":{"status":"ok","timestamp":1679184070474,"user_tz":-60,"elapsed":9,"user":{"displayName":"ALBERTO BARRERA HERRERO","userId":"09905877085308943679"}},"outputId":"a9e0e2b6-bfe6-403d-b07e-514558bc360c","id":"xz1YJD3nKw-6"},"outputs":[{"output_type":"stream","name":"stderr","text":["<ipython-input-31-bdf31e48ebde>:10: SettingWithCopyWarning: \n","A value is trying to be set on a copy of a slice from a DataFrame\n","\n","See the caveats in the documentation: https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#returning-a-view-versus-a-copy\n","  recom['artist_name'].iloc[i] = artist_mbid_2_name[artist_index2artist_mbid[recom['artist'].iloc[i]]][0]\n"]},{"output_type":"execute_result","data":{"text/plain":["      score      artist_name\n","0  1.000000         The Band\n","1  0.518695     Van Morrison\n","2  0.494562           Crosby\n","3  0.464024        Bob Dylan\n","4  0.462077           Crosby\n","5  0.435122    Grateful Dead\n","6  0.425560  George Harrison\n","7  0.420686       Neil Young\n","8  0.418821        The Byrds\n","9  0.413197     Warren Zevon"],"text/html":["\n","  <div id=\"df-bd5fe5e7-5d81-49c0-a9e4-bcdcbc07f565\">\n","    <div class=\"colab-df-container\">\n","      <div>\n","<style scoped>\n","    .dataframe tbody tr th:only-of-type {\n","        vertical-align: middle;\n","    }\n","\n","    .dataframe tbody tr th {\n","        vertical-align: top;\n","    }\n","\n","    .dataframe thead th {\n","        text-align: right;\n","    }\n","</style>\n","<table border=\"1\" class=\"dataframe\">\n","  <thead>\n","    <tr style=\"text-align: right;\">\n","      <th></th>\n","      <th>score</th>\n","      <th>artist_name</th>\n","    </tr>\n","  </thead>\n","  <tbody>\n","    <tr>\n","      <th>0</th>\n","      <td>1.000000</td>\n","      <td>The Band</td>\n","    </tr>\n","    <tr>\n","      <th>1</th>\n","      <td>0.518695</td>\n","      <td>Van Morrison</td>\n","    </tr>\n","    <tr>\n","      <th>2</th>\n","      <td>0.494562</td>\n","      <td>Crosby</td>\n","    </tr>\n","    <tr>\n","      <th>3</th>\n","      <td>0.464024</td>\n","      <td>Bob Dylan</td>\n","    </tr>\n","    <tr>\n","      <th>4</th>\n","      <td>0.462077</td>\n","      <td>Crosby</td>\n","    </tr>\n","    <tr>\n","      <th>5</th>\n","      <td>0.435122</td>\n","      <td>Grateful Dead</td>\n","    </tr>\n","    <tr>\n","      <th>6</th>\n","      <td>0.425560</td>\n","      <td>George Harrison</td>\n","    </tr>\n","    <tr>\n","      <th>7</th>\n","      <td>0.420686</td>\n","      <td>Neil Young</td>\n","    </tr>\n","    <tr>\n","      <th>8</th>\n","      <td>0.418821</td>\n","      <td>The Byrds</td>\n","    </tr>\n","    <tr>\n","      <th>9</th>\n","      <td>0.413197</td>\n","      <td>Warren Zevon</td>\n","    </tr>\n","  </tbody>\n","</table>\n","</div>\n","      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-bd5fe5e7-5d81-49c0-a9e4-bcdcbc07f565')\"\n","              title=\"Convert this dataframe to an interactive table.\"\n","              style=\"display:none;\">\n","        \n","  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n","       width=\"24px\">\n","    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n","    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n","  </svg>\n","      </button>\n","      \n","  <style>\n","    .colab-df-container {\n","      display:flex;\n","      flex-wrap:wrap;\n","      gap: 12px;\n","    }\n","\n","    .colab-df-convert {\n","      background-color: #E8F0FE;\n","      border: none;\n","      border-radius: 50%;\n","      cursor: pointer;\n","      display: none;\n","      fill: #1967D2;\n","      height: 32px;\n","      padding: 0 0 0 0;\n","      width: 32px;\n","    }\n","\n","    .colab-df-convert:hover {\n","      background-color: #E2EBFA;\n","      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n","      fill: #174EA6;\n","    }\n","\n","    [theme=dark] .colab-df-convert {\n","      background-color: #3B4455;\n","      fill: #D2E3FC;\n","    }\n","\n","    [theme=dark] .colab-df-convert:hover {\n","      background-color: #434B5C;\n","      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n","      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n","      fill: #FFFFFF;\n","    }\n","  </style>\n","\n","      <script>\n","        const buttonEl =\n","          document.querySelector('#df-bd5fe5e7-5d81-49c0-a9e4-bcdcbc07f565 button.colab-df-convert');\n","        buttonEl.style.display =\n","          google.colab.kernel.accessAllowed ? 'block' : 'none';\n","\n","        async function convertToInteractive(key) {\n","          const element = document.querySelector('#df-bd5fe5e7-5d81-49c0-a9e4-bcdcbc07f565');\n","          const dataTable =\n","            await google.colab.kernel.invokeFunction('convertToInteractive',\n","                                                     [key], {});\n","          if (!dataTable) return;\n","\n","          const docLinkHtml = 'Like what you see? Visit the ' +\n","            '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n","            + ' to learn more about interactive tables.';\n","          element.innerHTML = '';\n","          dataTable['output_type'] = 'display_data';\n","          await google.colab.output.renderOutput(dataTable, element);\n","          const docLink = document.createElement('div');\n","          docLink.innerHTML = docLinkHtml;\n","          element.appendChild(docLink);\n","        }\n","      </script>\n","    </div>\n","  </div>\n","  "]},"metadata":{},"execution_count":31}],"source":["# Check for The Band\n","id = artist_mbid_2_index[artist_name_2_mbid['The Band'][0]]\n","ids, scores= model.similar_items(id)\n","\n","# Display the results using pandas for nicer formatting\n","recom = pd.DataFrame({\"artist\": ids, \"score\": scores})\n","recom['artist_name'] = 0 * recom['artist']\n","\n","for i in range(len(recom['artist'])):\n","    recom['artist_name'].iloc[i] = artist_mbid_2_name[artist_index2artist_mbid[recom['artist'].iloc[i]]][0]\n","\n","recom = recom.drop(['artist'], axis=1)\n","recom"]}],"metadata":{"accelerator":"GPU","colab":{"provenance":[],"authorship_tag":"ABX9TyPkJIlPLEw/JpzOzE0cnQCW"},"gpuClass":"standard","kernelspec":{"display_name":"Python 3","name":"python3"},"language_info":{"name":"python"},"widgets":{"application/vnd.jupyter.widget-state+json":{"c1e146def1d44e40bdfadaa22e4f31aa":{"model_module":"@jupyter-widgets/controls","model_name":"HBoxModel","model_module_version":"1.5.0","state":{"_dom_classes":[],"_model_module":"@jupyter-widgets/controls","_model_module_version":"1.5.0","_model_name":"HBoxModel","_view_count":null,"_view_module":"@jupyter-widgets/controls","_view_module_version":"1.5.0","_view_name":"HBoxView","box_style":"","children":["IPY_MODEL_d1173548d5db4ece9c6db841cea9f4eb","IPY_MODEL_04741fb123ae499c870c105f5f6d08a8","IPY_MODEL_59a5360c93064f61905f9f8abc0aa3d8"],"layout":"IPY_MODEL_323689fcce7e4dc5b1ec17f638381a37"}},"d1173548d5db4ece9c6db841cea9f4eb":{"model_module":"@jupyter-widgets/controls","model_name":"HTMLModel","model_module_version":"1.5.0","state":{"_dom_classes":[],"_model_module":"@jupyter-widgets/controls","_model_module_version":"1.5.0","_model_name":"HTMLModel","_view_count":null,"_view_module":"@jupyter-widgets/controls","_view_module_version":"1.5.0","_view_name":"HTMLView","description":"","description_tooltip":null,"layout":"IPY_MODEL_06a8eda9faa94df9adc394505f18df9e","placeholder":"​","style":"IPY_MODEL_1120c521aa7c43caa50bf32dc74e1331","value":"100%"}},"04741fb123ae499c870c105f5f6d08a8":{"model_module":"@jupyter-widgets/controls","model_name":"FloatProgressModel","model_module_version":"1.5.0","state":{"_dom_classes":[],"_model_module":"@jupyter-widgets/controls","_model_module_version":"1.5.0","_model_name":"FloatProgressModel","_view_count":null,"_view_module":"@jupyter-widgets/controls","_view_module_version":"1.5.0","_view_name":"ProgressView","bar_style":"success","description":"","description_tooltip":null,"layout":"IPY_MODEL_61c65e39ed164a579bf33b8fadcb40b5","max":15,"min":0,"orientation":"horizontal","style":"IPY_MODEL_2a6638270cc44ff2983ce87308e5802d","value":15}},"59a5360c93064f61905f9f8abc0aa3d8":{"model_module":"@jupyter-widgets/controls","model_name":"HTMLModel","model_module_version":"1.5.0","state":{"_dom_classes":[],"_model_module":"@jupyter-widgets/controls","_model_module_version":"1.5.0","_model_name":"HTMLModel","_view_count":null,"_view_module":"@jupyter-widgets/controls","_view_module_version":"1.5.0","_view_name":"HTMLView","description":"","description_tooltip":null,"layout":"IPY_MODEL_44e4c4f101de422d9ad53ecc8b3e9c5f","placeholder":"​","style":"IPY_MODEL_97739611e8f74363b0d929adc5dccaae","value":" 15/15 [01:45&lt;00:00,  6.83s/it]"}},"323689fcce7e4dc5b1ec17f638381a37":{"model_module":"@jupyter-widgets/base","model_name":"LayoutModel","model_module_version":"1.2.0","state":{"_model_module":"@jupyter-widgets/base","_model_module_version":"1.2.0","_model_name":"LayoutModel","_view_count":null,"_view_module":"@jupyter-widgets/base","_view_module_version":"1.2.0","_view_name":"LayoutView","align_content":null,"align_items":null,"align_self":null,"border":null,"bottom":null,"display":null,"flex":null,"flex_flow":null,"grid_area":null,"grid_auto_columns":null,"grid_auto_flow":null,"grid_auto_rows":null,"grid_column":null,"grid_gap":null,"grid_row":null,"grid_template_areas":null,"grid_template_columns":null,"grid_template_rows":null,"height":null,"justify_content":null,"justify_items":null,"left":null,"margin":null,"max_height":null,"max_width":null,"min_height":null,"min_width":null,"object_fit":null,"object_position":null,"order":null,"overflow":null,"overflow_x":null,"overflow_y":null,"padding":null,"right":null,"top":null,"visibility":null,"width":null}},"06a8eda9faa94df9adc394505f18df9e":{"model_module":"@jupyter-widgets/base","model_name":"LayoutModel","model_module_version":"1.2.0","state":{"_model_module":"@jupyter-widgets/base","_model_module_version":"1.2.0","_model_name":"LayoutModel","_view_count":null,"_view_module":"@jupyter-widgets/base","_view_module_version":"1.2.0","_view_name":"LayoutView","align_content":null,"align_items":null,"align_self":null,"border":null,"bottom":null,"display":null,"flex":null,"flex_flow":null,"grid_area":null,"grid_auto_columns":null,"grid_auto_flow":null,"grid_auto_rows":null,"grid_column":null,"grid_gap":null,"grid_row":null,"grid_template_areas":null,"grid_template_columns":null,"grid_template_rows":null,"height":null,"justify_content":null,"justify_items":null,"left":null,"margin":null,"max_height":null,"max_width":null,"min_height":null,"min_width":null,"object_fit":null,"object_position":null,"order":null,"overflow":null,"overflow_x":null,"overflow_y":null,"padding":null,"right":null,"top":null,"visibility":null,"width":null}},"1120c521aa7c43caa50bf32dc74e1331":{"model_module":"@jupyter-widgets/controls","model_name":"DescriptionStyleModel","model_module_version":"1.5.0","state":{"_model_module":"@jupyter-widgets/controls","_model_module_version":"1.5.0","_model_name":"DescriptionStyleModel","_view_count":null,"_view_module":"@jupyter-widgets/base","_view_module_version":"1.2.0","_view_name":"StyleView","description_width":""}},"61c65e39ed164a579bf33b8fadcb40b5":{"model_module":"@jupyter-widgets/base","model_name":"LayoutModel","model_module_version":"1.2.0","state":{"_model_module":"@jupyter-widgets/base","_model_module_version":"1.2.0","_model_name":"LayoutModel","_view_count":null,"_view_module":"@jupyter-widgets/base","_view_module_version":"1.2.0","_view_name":"LayoutView","align_content":null,"align_items":null,"align_self":null,"border":null,"bottom":null,"display":null,"flex":null,"flex_flow":null,"grid_area":null,"grid_auto_columns":null,"grid_auto_flow":null,"grid_auto_rows":null,"grid_column":null,"grid_gap":null,"grid_row":null,"grid_template_areas":null,"grid_template_columns":null,"grid_template_rows":null,"height":null,"justify_content":null,"justify_items":null,"left":null,"margin":null,"max_height":null,"max_width":null,"min_height":null,"min_width":null,"object_fit":null,"object_position":null,"order":null,"overflow":null,"overflow_x":null,"overflow_y":null,"padding":null,"right":null,"top":null,"visibility":null,"width":null}},"2a6638270cc44ff2983ce87308e5802d":{"model_module":"@jupyter-widgets/controls","model_name":"ProgressStyleModel","model_module_version":"1.5.0","state":{"_model_module":"@jupyter-widgets/controls","_model_module_version":"1.5.0","_model_name":"ProgressStyleModel","_view_count":null,"_view_module":"@jupyter-widgets/base","_view_module_version":"1.2.0","_view_name":"StyleView","bar_color":null,"description_width":""}},"44e4c4f101de422d9ad53ecc8b3e9c5f":{"model_module":"@jupyter-widgets/base","model_name":"LayoutModel","model_module_version":"1.2.0","state":{"_model_module":"@jupyter-widgets/base","_model_module_version":"1.2.0","_model_name":"LayoutModel","_view_count":null,"_view_module":"@jupyter-widgets/base","_view_module_version":"1.2.0","_view_name":"LayoutView","align_content":null,"align_items":null,"align_self":null,"border":null,"bottom":null,"display":null,"flex":null,"flex_flow":null,"grid_area":null,"grid_auto_columns":null,"grid_auto_flow":null,"grid_auto_rows":null,"grid_column":null,"grid_gap":null,"grid_row":null,"grid_template_areas":null,"grid_template_columns":null,"grid_template_rows":null,"height":null,"justify_content":null,"justify_items":null,"left":null,"margin":null,"max_height":null,"max_width":null,"min_height":null,"min_width":null,"object_fit":null,"object_position":null,"order":null,"overflow":null,"overflow_x":null,"overflow_y":null,"padding":null,"right":null,"top":null,"visibility":null,"width":null}},"97739611e8f74363b0d929adc5dccaae":{"model_module":"@jupyter-widgets/controls","model_name":"DescriptionStyleModel","model_module_version":"1.5.0","state":{"_model_module":"@jupyter-widgets/controls","_model_module_version":"1.5.0","_model_name":"DescriptionStyleModel","_view_count":null,"_view_module":"@jupyter-widgets/base","_view_module_version":"1.2.0","_view_name":"StyleView","description_width":""}}}}},"nbformat":4,"nbformat_minor":0}