code/04_recommeder_sys.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "![](../images/bunker_studer.jpeg)\n",
    "<br>\n",
    "Photo: [*The Bunker*](https://www.thebunkerstudio.com/)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# The Sound of Science\n",
    "### Notebook 4: Recommender System\n",
    "#### Adam Zucker\n",
    "---"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Contents\n",
    "- **Section 1:** Package and data imports, preprocessing\n",
    "- **Section 2:** Vector generation and dataframe creation\n",
    "- **Section 3:** Recommender algorithm"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "---\n",
    "---\n",
    "### Section 1\n",
    "#### Imports"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import pandas as pd\n",
    "import matplotlib.pyplot as plt\n",
    "import seaborn as sns\n",
    "import scipy as sp\n",
    "\n",
    "from sklearn.preprocessing import StandardScaler, OneHotEncoder\n",
    "from sklearn.metrics.pairwise import pairwise_distances, cosine_distances, cosine_similarity, manhattan_distances\n",
    "\n",
    "import librosa as lib\n",
    "import librosa.display as libd"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<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>name</th>\n",
       "      <th>artists</th>\n",
       "      <th>tempo</th>\n",
       "      <th>key</th>\n",
       "      <th>mode</th>\n",
       "      <th>full_key</th>\n",
       "      <th>A minor</th>\n",
       "      <th>A# major</th>\n",
       "      <th>A# minor</th>\n",
       "      <th>B major</th>\n",
       "      <th>...</th>\n",
       "      <th>energy</th>\n",
       "      <th>instrumentalness</th>\n",
       "      <th>speechiness</th>\n",
       "      <th>acousticness</th>\n",
       "      <th>danceability</th>\n",
       "      <th>valence</th>\n",
       "      <th>popularity</th>\n",
       "      <th>liveness</th>\n",
       "      <th>year</th>\n",
       "      <th>id</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Thunderstruck</td>\n",
       "      <td>['AC/DC']</td>\n",
       "      <td>133.5</td>\n",
       "      <td>4</td>\n",
       "      <td>1</td>\n",
       "      <td>E major</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0.890</td>\n",
       "      <td>0.01170</td>\n",
       "      <td>0.0364</td>\n",
       "      <td>0.000147</td>\n",
       "      <td>0.502</td>\n",
       "      <td>0.259</td>\n",
       "      <td>83</td>\n",
       "      <td>0.217</td>\n",
       "      <td>1990</td>\n",
       "      <td>57bgtoPSgt236HzfBOd8kj</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>The Gift of Love</td>\n",
       "      <td>['Bette Midler']</td>\n",
       "      <td>157.5</td>\n",
       "      <td>8</td>\n",
       "      <td>1</td>\n",
       "      <td>G# major</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0.467</td>\n",
       "      <td>0.00000</td>\n",
       "      <td>0.0287</td>\n",
       "      <td>0.359000</td>\n",
       "      <td>0.486</td>\n",
       "      <td>0.286</td>\n",
       "      <td>38</td>\n",
       "      <td>0.110</td>\n",
       "      <td>1990</td>\n",
       "      <td>7FUc1xVSKvABmVwI6kS5Y4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Thelma - Bonus Track</td>\n",
       "      <td>['Paul Simon']</td>\n",
       "      <td>94.0</td>\n",
       "      <td>5</td>\n",
       "      <td>1</td>\n",
       "      <td>F major</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0.529</td>\n",
       "      <td>0.08450</td>\n",
       "      <td>0.0770</td>\n",
       "      <td>0.872000</td>\n",
       "      <td>0.710</td>\n",
       "      <td>0.882</td>\n",
       "      <td>29</td>\n",
       "      <td>0.093</td>\n",
       "      <td>1990</td>\n",
       "      <td>7pcEC5r1jVqWGRypo9D7f7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>How I Need You</td>\n",
       "      <td>['Bad Boys Blue']</td>\n",
       "      <td>123.1</td>\n",
       "      <td>9</td>\n",
       "      <td>0</td>\n",
       "      <td>A minor</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0.670</td>\n",
       "      <td>0.00347</td>\n",
       "      <td>0.0398</td>\n",
       "      <td>0.072400</td>\n",
       "      <td>0.652</td>\n",
       "      <td>0.963</td>\n",
       "      <td>44</td>\n",
       "      <td>0.119</td>\n",
       "      <td>1990</td>\n",
       "      <td>1yq8h4zD0IDT5X1YTaEwZh</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Nunca Dudes De Mi</td>\n",
       "      <td>['El Golpe']</td>\n",
       "      <td>143.1</td>\n",
       "      <td>4</td>\n",
       "      <td>1</td>\n",
       "      <td>E major</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0.490</td>\n",
       "      <td>0.00000</td>\n",
       "      <td>0.0295</td>\n",
       "      <td>0.151000</td>\n",
       "      <td>0.476</td>\n",
       "      <td>0.514</td>\n",
       "      <td>31</td>\n",
       "      <td>0.305</td>\n",
       "      <td>1990</td>\n",
       "      <td>5kNYkLFs3WFFgE6qhfWDEm</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 41 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                   name            artists  tempo  key  mode  full_key  \\\n",
       "0         Thunderstruck          ['AC/DC']  133.5    4     1   E major   \n",
       "1      The Gift of Love   ['Bette Midler']  157.5    8     1  G# major   \n",
       "2  Thelma - Bonus Track     ['Paul Simon']   94.0    5     1   F major   \n",
       "3        How I Need You  ['Bad Boys Blue']  123.1    9     0   A minor   \n",
       "4     Nunca Dudes De Mi       ['El Golpe']  143.1    4     1   E major   \n",
       "\n",
       "   A minor  A# major  A# minor  B major  ...  energy  instrumentalness  \\\n",
       "0        0         0         0        0  ...   0.890           0.01170   \n",
       "1        0         0         0        0  ...   0.467           0.00000   \n",
       "2        0         0         0        0  ...   0.529           0.08450   \n",
       "3        1         0         0        0  ...   0.670           0.00347   \n",
       "4        0         0         0        0  ...   0.490           0.00000   \n",
       "\n",
       "   speechiness  acousticness  danceability  valence  popularity  liveness  \\\n",
       "0       0.0364      0.000147         0.502    0.259          83     0.217   \n",
       "1       0.0287      0.359000         0.486    0.286          38     0.110   \n",
       "2       0.0770      0.872000         0.710    0.882          29     0.093   \n",
       "3       0.0398      0.072400         0.652    0.963          44     0.119   \n",
       "4       0.0295      0.151000         0.476    0.514          31     0.305   \n",
       "\n",
       "   year                      id  \n",
       "0  1990  57bgtoPSgt236HzfBOd8kj  \n",
       "1  1990  7FUc1xVSKvABmVwI6kS5Y4  \n",
       "2  1990  7pcEC5r1jVqWGRypo9D7f7  \n",
       "3  1990  1yq8h4zD0IDT5X1YTaEwZh  \n",
       "4  1990  5kNYkLFs3WFFgE6qhfWDEm  \n",
       "\n",
       "[5 rows x 41 columns]"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Importing cleaned Spotify dataframe\n",
    "df = pd.read_csv('../data_clean/spotify_kg_master.csv')\n",
    "dummy_df = pd.read_csv('../data_clean/spotify_kg_dummied.csv')\n",
    "df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "---"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "**BELOW:** Preprocessing and formatting the Spotify dataframe for conversion to a sparse matrix."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(56798, 41)\n",
      "(56798, 41)\n"
     ]
    }
   ],
   "source": [
    "# Creating a copy of the dataframe to use for the recommender system\n",
    "temp_df = df.copy()\n",
    "print(temp_df.shape)\n",
    "print(df.shape)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Index(['name', 'artists', 'tempo', 'key', 'mode', 'full_key', 'A minor',\n",
       "       'A# major', 'A# minor', 'B major', 'B minor', 'C major', 'C minor',\n",
       "       'C# major', 'C# minor', 'D major', 'D minor', 'D# major', 'D# minor',\n",
       "       'E major', 'E minor', 'F major', 'F minor', 'F# major', 'F# minor',\n",
       "       'G major', 'G minor', 'G# major', 'G# minor', 'loudness', 'duration_s',\n",
       "       'energy', 'instrumentalness', 'speechiness', 'acousticness',\n",
       "       'danceability', 'valence', 'popularity', 'liveness', 'year', 'id'],\n",
       "      dtype='object')"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "temp_df.columns"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Combining 'name' and 'artists' features to use as indices\n",
    "temp_df['name_and_artists'] = temp_df['name'] + ' - ' + temp_df['artists']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Dropping features that won't be used for similarity comparisons against new, incoming songs\n",
    "temp_df.drop(columns=['name', 'artists', 'full_key', 'secondary_full_key', 'popularity', 'year', 'id'], inplace=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Index(['tempo', 'A minor', 'A# major', 'A# minor', 'B major', 'B minor',\n",
       "       'C major', 'C minor', 'C# major', 'C# minor', 'D major', 'D minor',\n",
       "       'D# major', 'D# minor', 'E major', 'E minor', 'F major', 'F minor',\n",
       "       'F# major', 'F# minor', 'G major', 'G minor', 'G# major', 'G# minor',\n",
       "       'loudness', 'duration_s', 'energy', 'instrumentalness', 'speechiness',\n",
       "       'acousticness', 'danceability', 'valence', 'liveness',\n",
       "       'name_and_artists'],\n",
       "      dtype='object')"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "temp_df.columns"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Reordering columns\n",
    "temp_df = temp_df[['name_and_artists', 'tempo', 'key', 'mode', 'secondary_key', 'secondary_mode', 'loudness', \n",
    "                   'duration_s', 'energy', 'instrumentalness', 'speechiness', 'acousticness', 'danceability', \n",
    "                   'valence', 'liveness']]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "# # Reordering columns\n",
    "# temp_df = temp_df[['name_and_artists', 'tempo', 'A minor', 'A# major', 'A# minor', 'B major', 'B minor', 'C major', \n",
    "#                    'C minor', 'C# major', 'C# minor', 'D major', 'D minor', 'D# major', 'D# minor', 'E major', 'E minor', \n",
    "#                    'F major', 'F minor', 'F# major', 'F# minor', 'G major', 'G minor', 'G# major', 'G# minor', 'loudness', \n",
    "#                    'duration_s', 'energy', 'instrumentalness', 'speechiness', 'acousticness', 'danceability', 'valence', \n",
    "#                    'liveness']]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Setting the song titles and associated artists as the index in my new dataframe\n",
    "temp_df.set_index(keys='name_and_artists', inplace=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<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>tempo</th>\n",
       "      <th>A minor</th>\n",
       "      <th>A# major</th>\n",
       "      <th>A# minor</th>\n",
       "      <th>B major</th>\n",
       "      <th>B minor</th>\n",
       "      <th>C major</th>\n",
       "      <th>C minor</th>\n",
       "      <th>C# major</th>\n",
       "      <th>C# minor</th>\n",
       "      <th>...</th>\n",
       "      <th>G# minor</th>\n",
       "      <th>loudness</th>\n",
       "      <th>duration_s</th>\n",
       "      <th>energy</th>\n",
       "      <th>instrumentalness</th>\n",
       "      <th>speechiness</th>\n",
       "      <th>acousticness</th>\n",
       "      <th>danceability</th>\n",
       "      <th>valence</th>\n",
       "      <th>liveness</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>name_and_artists</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>Thunderstruck - ['AC/DC']</th>\n",
       "      <td>133.5</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>-5.175</td>\n",
       "      <td>292.9</td>\n",
       "      <td>0.890</td>\n",
       "      <td>0.01170</td>\n",
       "      <td>0.0364</td>\n",
       "      <td>0.000147</td>\n",
       "      <td>0.502</td>\n",
       "      <td>0.259</td>\n",
       "      <td>0.217</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>The Gift of Love - ['Bette Midler']</th>\n",
       "      <td>157.5</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>-10.765</td>\n",
       "      <td>241.6</td>\n",
       "      <td>0.467</td>\n",
       "      <td>0.00000</td>\n",
       "      <td>0.0287</td>\n",
       "      <td>0.359000</td>\n",
       "      <td>0.486</td>\n",
       "      <td>0.286</td>\n",
       "      <td>0.110</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Thelma - Bonus Track - ['Paul Simon']</th>\n",
       "      <td>94.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>-13.367</td>\n",
       "      <td>254.7</td>\n",
       "      <td>0.529</td>\n",
       "      <td>0.08450</td>\n",
       "      <td>0.0770</td>\n",
       "      <td>0.872000</td>\n",
       "      <td>0.710</td>\n",
       "      <td>0.882</td>\n",
       "      <td>0.093</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>How I Need You - ['Bad Boys Blue']</th>\n",
       "      <td>123.1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>-12.096</td>\n",
       "      <td>218.3</td>\n",
       "      <td>0.670</td>\n",
       "      <td>0.00347</td>\n",
       "      <td>0.0398</td>\n",
       "      <td>0.072400</td>\n",
       "      <td>0.652</td>\n",
       "      <td>0.963</td>\n",
       "      <td>0.119</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Nunca Dudes De Mi - ['El Golpe']</th>\n",
       "      <td>143.1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>-8.606</td>\n",
       "      <td>176.0</td>\n",
       "      <td>0.490</td>\n",
       "      <td>0.00000</td>\n",
       "      <td>0.0295</td>\n",
       "      <td>0.151000</td>\n",
       "      <td>0.476</td>\n",
       "      <td>0.514</td>\n",
       "      <td>0.305</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 33 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                                       tempo  A minor  A# major  A# minor  \\\n",
       "name_and_artists                                                            \n",
       "Thunderstruck - ['AC/DC']              133.5        0         0         0   \n",
       "The Gift of Love - ['Bette Midler']    157.5        0         0         0   \n",
       "Thelma - Bonus Track - ['Paul Simon']   94.0        0         0         0   \n",
       "How I Need You - ['Bad Boys Blue']     123.1        1         0         0   \n",
       "Nunca Dudes De Mi - ['El Golpe']       143.1        0         0         0   \n",
       "\n",
       "                                       B major  B minor  C major  C minor  \\\n",
       "name_and_artists                                                            \n",
       "Thunderstruck - ['AC/DC']                    0        0        0        0   \n",
       "The Gift of Love - ['Bette Midler']          0        0        0        0   \n",
       "Thelma - Bonus Track - ['Paul Simon']        0        0        0        0   \n",
       "How I Need You - ['Bad Boys Blue']           0        0        0        0   \n",
       "Nunca Dudes De Mi - ['El Golpe']             0        0        0        0   \n",
       "\n",
       "                                       C# major  C# minor  ...  G# minor  \\\n",
       "name_and_artists                                           ...             \n",
       "Thunderstruck - ['AC/DC']                     0         0  ...         0   \n",
       "The Gift of Love - ['Bette Midler']           0         0  ...         0   \n",
       "Thelma - Bonus Track - ['Paul Simon']         0         0  ...         0   \n",
       "How I Need You - ['Bad Boys Blue']            0         0  ...         0   \n",
       "Nunca Dudes De Mi - ['El Golpe']              0         0  ...         0   \n",
       "\n",
       "                                       loudness  duration_s  energy  \\\n",
       "name_and_artists                                                      \n",
       "Thunderstruck - ['AC/DC']                -5.175       292.9   0.890   \n",
       "The Gift of Love - ['Bette Midler']     -10.765       241.6   0.467   \n",
       "Thelma - Bonus Track - ['Paul Simon']   -13.367       254.7   0.529   \n",
       "How I Need You - ['Bad Boys Blue']      -12.096       218.3   0.670   \n",
       "Nunca Dudes De Mi - ['El Golpe']         -8.606       176.0   0.490   \n",
       "\n",
       "                                       instrumentalness  speechiness  \\\n",
       "name_and_artists                                                       \n",
       "Thunderstruck - ['AC/DC']                       0.01170       0.0364   \n",
       "The Gift of Love - ['Bette Midler']             0.00000       0.0287   \n",
       "Thelma - Bonus Track - ['Paul Simon']           0.08450       0.0770   \n",
       "How I Need You - ['Bad Boys Blue']              0.00347       0.0398   \n",
       "Nunca Dudes De Mi - ['El Golpe']                0.00000       0.0295   \n",
       "\n",
       "                                       acousticness  danceability  valence  \\\n",
       "name_and_artists                                                             \n",
       "Thunderstruck - ['AC/DC']                  0.000147         0.502    0.259   \n",
       "The Gift of Love - ['Bette Midler']        0.359000         0.486    0.286   \n",
       "Thelma - Bonus Track - ['Paul Simon']      0.872000         0.710    0.882   \n",
       "How I Need You - ['Bad Boys Blue']         0.072400         0.652    0.963   \n",
       "Nunca Dudes De Mi - ['El Golpe']           0.151000         0.476    0.514   \n",
       "\n",
       "                                       liveness  \n",
       "name_and_artists                                 \n",
       "Thunderstruck - ['AC/DC']                 0.217  \n",
       "The Gift of Love - ['Bette Midler']       0.110  \n",
       "Thelma - Bonus Track - ['Paul Simon']     0.093  \n",
       "How I Need You - ['Bad Boys Blue']        0.119  \n",
       "Nunca Dudes De Mi - ['El Golpe']          0.305  \n",
       "\n",
       "[5 rows x 33 columns]"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "temp_df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "---\n",
    "---\n",
    "### Section 2\n",
    "#### Vector Generation"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Instantiating Standard Scaler and fitting/transforming my dataframe\n",
    "ss = StandardScaler()\n",
    "\n",
    "scaled_df = ss.fit_transform(temp_df)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array([[ 0.41221902, -0.17997382, -0.19574651, ..., -0.50325348,\n",
       "        -1.01862261,  0.27689627],\n",
       "       [ 1.19629352, -0.17997382, -0.19574651, ..., -0.5957251 ,\n",
       "        -0.9125911 , -0.52194792],\n",
       "       [-0.87823693, -0.17997382, -0.19574651, ...,  0.69887759,\n",
       "         1.42795612, -0.64886709],\n",
       "       ...,\n",
       "       [-0.29018105, -0.17997382, -0.19574651, ...,  2.11484929,\n",
       "         1.50257088,  3.024323  ],\n",
       "       [ 0.29787482, -0.17997382, -0.19574651, ...,  0.13248891,\n",
       "         0.30480762, -0.21584875],\n",
       "       [-0.84556716, -0.17997382, -0.19574651, ...,  1.54268113,\n",
       "         0.48545388,  0.01559209]])"
      ]
     },
     "execution_count": 24,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "scaled_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [],
   "source": [
    "# rec = pairwise_distances(scaled_df, metric='cosine', n_jobs=12)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [],
   "source": [
    "rec = 1 - cosine_distances(scaled_df)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "rec_df = pd.DataFrame(data=rec, columns=temp_df.index, index=temp_df.index)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<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>name_and_artists</th>\n",
       "      <th>Thunderstruck - ['AC/DC']</th>\n",
       "      <th>The Gift of Love - ['Bette Midler']</th>\n",
       "      <th>Thelma - Bonus Track - ['Paul Simon']</th>\n",
       "      <th>How I Need You - ['Bad Boys Blue']</th>\n",
       "      <th>Nunca Dudes De Mi - ['El Golpe']</th>\n",
       "      <th>You And Me Of The 10,000 Wars - ['Indigo Girls']</th>\n",
       "      <th>Don't Laugh (i Love You) - ['Ween']</th>\n",
       "      <th>Railroad Worksong - ['The Notting Hillbillies']</th>\n",
       "      <th>Turn! Turn! Turn! - ['The Byrds']</th>\n",
       "      <th>Siéntate Ahí - [\"Oscar D'León\"]</th>\n",
       "      <th>...</th>\n",
       "      <th>I WANNA SEE SOME ASS (feat. jetsonmade) - ['Jack Harlow', 'jetsonmade']</th>\n",
       "      <th>MODUS - ['Joji']</th>\n",
       "      <th>Psycho! - ['MASN']</th>\n",
       "      <th>COOLER THAN A BITCH (feat. Roddy Ricch) - ['Gunna', 'Roddy Ricch']</th>\n",
       "      <th>What If I Told You That I Love You - ['Ali Gatie']</th>\n",
       "      <th>24 (feat. Lil Baby) - ['Money Man', 'Lil Baby']</th>\n",
       "      <th>Bean (Kobe) [feat. Chief Keef] - ['Lil Uzi Vert', 'Chief Keef']</th>\n",
       "      <th>Pardon (feat. Lil Baby) - ['T.I.', 'Lil Baby']</th>\n",
       "      <th>Long Live - ['Florida Georgia Line']</th>\n",
       "      <th>Billetes Azules (with J Balvin) - ['KEVVO', 'J Balvin']</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>name_and_artists</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>Thunderstruck - ['AC/DC']</th>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.008766</td>\n",
       "      <td>1.200292</td>\n",
       "      <td>1.081286</td>\n",
       "      <td>1.642551e-01</td>\n",
       "      <td>1.124453</td>\n",
       "      <td>1.059340</td>\n",
       "      <td>1.053503</td>\n",
       "      <td>1.039069</td>\n",
       "      <td>1.010914</td>\n",
       "      <td>...</td>\n",
       "      <td>1.193503</td>\n",
       "      <td>1.037507</td>\n",
       "      <td>1.180821</td>\n",
       "      <td>1.108988</td>\n",
       "      <td>1.186523</td>\n",
       "      <td>1.097103</td>\n",
       "      <td>1.048400</td>\n",
       "      <td>1.078885</td>\n",
       "      <td>1.027577</td>\n",
       "      <td>1.103890</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>The Gift of Love - ['Bette Midler']</th>\n",
       "      <td>1.008766</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.071814</td>\n",
       "      <td>1.067803</td>\n",
       "      <td>9.874569e-01</td>\n",
       "      <td>0.935461</td>\n",
       "      <td>1.236937</td>\n",
       "      <td>0.989657</td>\n",
       "      <td>1.030825</td>\n",
       "      <td>1.156865</td>\n",
       "      <td>...</td>\n",
       "      <td>1.191703</td>\n",
       "      <td>0.980848</td>\n",
       "      <td>1.029696</td>\n",
       "      <td>1.081106</td>\n",
       "      <td>0.990258</td>\n",
       "      <td>1.083273</td>\n",
       "      <td>1.216447</td>\n",
       "      <td>1.199216</td>\n",
       "      <td>1.116030</td>\n",
       "      <td>1.238159</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Thelma - Bonus Track - ['Paul Simon']</th>\n",
       "      <td>1.200292</td>\n",
       "      <td>1.071814</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.956827</td>\n",
       "      <td>1.127582e+00</td>\n",
       "      <td>0.936057</td>\n",
       "      <td>0.973261</td>\n",
       "      <td>0.929090</td>\n",
       "      <td>1.028249</td>\n",
       "      <td>0.984139</td>\n",
       "      <td>...</td>\n",
       "      <td>0.959309</td>\n",
       "      <td>1.138327</td>\n",
       "      <td>0.954939</td>\n",
       "      <td>0.973634</td>\n",
       "      <td>0.906705</td>\n",
       "      <td>1.065129</td>\n",
       "      <td>1.137969</td>\n",
       "      <td>1.009152</td>\n",
       "      <td>1.163957</td>\n",
       "      <td>1.043010</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>How I Need You - ['Bad Boys Blue']</th>\n",
       "      <td>1.081286</td>\n",
       "      <td>1.067803</td>\n",
       "      <td>0.956827</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.027095e+00</td>\n",
       "      <td>1.076275</td>\n",
       "      <td>0.777341</td>\n",
       "      <td>0.984242</td>\n",
       "      <td>0.907169</td>\n",
       "      <td>0.963341</td>\n",
       "      <td>...</td>\n",
       "      <td>0.894979</td>\n",
       "      <td>1.087290</td>\n",
       "      <td>1.064838</td>\n",
       "      <td>0.982186</td>\n",
       "      <td>1.123387</td>\n",
       "      <td>1.006332</td>\n",
       "      <td>1.132095</td>\n",
       "      <td>1.011595</td>\n",
       "      <td>1.011847</td>\n",
       "      <td>0.994169</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Nunca Dudes De Mi - ['El Golpe']</th>\n",
       "      <td>0.164255</td>\n",
       "      <td>0.987457</td>\n",
       "      <td>1.127582</td>\n",
       "      <td>1.027095</td>\n",
       "      <td>3.330669e-16</td>\n",
       "      <td>1.042554</td>\n",
       "      <td>1.010274</td>\n",
       "      <td>1.079293</td>\n",
       "      <td>1.044738</td>\n",
       "      <td>1.069386</td>\n",
       "      <td>...</td>\n",
       "      <td>1.041934</td>\n",
       "      <td>1.043607</td>\n",
       "      <td>1.021067</td>\n",
       "      <td>1.060800</td>\n",
       "      <td>1.098322</td>\n",
       "      <td>1.080464</td>\n",
       "      <td>1.094262</td>\n",
       "      <td>1.040128</td>\n",
       "      <td>1.003542</td>\n",
       "      <td>1.110944</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 56798 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "name_and_artists                       Thunderstruck - ['AC/DC']  \\\n",
       "name_and_artists                                                   \n",
       "Thunderstruck - ['AC/DC']                               0.000000   \n",
       "The Gift of Love - ['Bette Midler']                     1.008766   \n",
       "Thelma - Bonus Track - ['Paul Simon']                   1.200292   \n",
       "How I Need You - ['Bad Boys Blue']                      1.081286   \n",
       "Nunca Dudes De Mi - ['El Golpe']                        0.164255   \n",
       "\n",
       "name_and_artists                       The Gift of Love - ['Bette Midler']  \\\n",
       "name_and_artists                                                             \n",
       "Thunderstruck - ['AC/DC']                                         1.008766   \n",
       "The Gift of Love - ['Bette Midler']                               0.000000   \n",
       "Thelma - Bonus Track - ['Paul Simon']                             1.071814   \n",
       "How I Need You - ['Bad Boys Blue']                                1.067803   \n",
       "Nunca Dudes De Mi - ['El Golpe']                                  0.987457   \n",
       "\n",
       "name_and_artists                       Thelma - Bonus Track - ['Paul Simon']  \\\n",
       "name_and_artists                                                               \n",
       "Thunderstruck - ['AC/DC']                                           1.200292   \n",
       "The Gift of Love - ['Bette Midler']                                 1.071814   \n",
       "Thelma - Bonus Track - ['Paul Simon']                               0.000000   \n",
       "How I Need You - ['Bad Boys Blue']                                  0.956827   \n",
       "Nunca Dudes De Mi - ['El Golpe']                                    1.127582   \n",
       "\n",
       "name_and_artists                       How I Need You - ['Bad Boys Blue']  \\\n",
       "name_and_artists                                                            \n",
       "Thunderstruck - ['AC/DC']                                        1.081286   \n",
       "The Gift of Love - ['Bette Midler']                              1.067803   \n",
       "Thelma - Bonus Track - ['Paul Simon']                            0.956827   \n",
       "How I Need You - ['Bad Boys Blue']                               0.000000   \n",
       "Nunca Dudes De Mi - ['El Golpe']                                 1.027095   \n",
       "\n",
       "name_and_artists                       Nunca Dudes De Mi - ['El Golpe']  \\\n",
       "name_and_artists                                                          \n",
       "Thunderstruck - ['AC/DC']                                  1.642551e-01   \n",
       "The Gift of Love - ['Bette Midler']                        9.874569e-01   \n",
       "Thelma - Bonus Track - ['Paul Simon']                      1.127582e+00   \n",
       "How I Need You - ['Bad Boys Blue']                         1.027095e+00   \n",
       "Nunca Dudes De Mi - ['El Golpe']                           3.330669e-16   \n",
       "\n",
       "name_and_artists                       You And Me Of The 10,000 Wars - ['Indigo Girls']  \\\n",
       "name_and_artists                                                                          \n",
       "Thunderstruck - ['AC/DC']                                                      1.124453   \n",
       "The Gift of Love - ['Bette Midler']                                            0.935461   \n",
       "Thelma - Bonus Track - ['Paul Simon']                                          0.936057   \n",
       "How I Need You - ['Bad Boys Blue']                                             1.076275   \n",
       "Nunca Dudes De Mi - ['El Golpe']                                               1.042554   \n",
       "\n",
       "name_and_artists                       Don't Laugh (i Love You) - ['Ween']  \\\n",
       "name_and_artists                                                             \n",
       "Thunderstruck - ['AC/DC']                                         1.059340   \n",
       "The Gift of Love - ['Bette Midler']                               1.236937   \n",
       "Thelma - Bonus Track - ['Paul Simon']                             0.973261   \n",
       "How I Need You - ['Bad Boys Blue']                                0.777341   \n",
       "Nunca Dudes De Mi - ['El Golpe']                                  1.010274   \n",
       "\n",
       "name_and_artists                       Railroad Worksong - ['The Notting Hillbillies']  \\\n",
       "name_and_artists                                                                         \n",
       "Thunderstruck - ['AC/DC']                                                     1.053503   \n",
       "The Gift of Love - ['Bette Midler']                                           0.989657   \n",
       "Thelma - Bonus Track - ['Paul Simon']                                         0.929090   \n",
       "How I Need You - ['Bad Boys Blue']                                            0.984242   \n",
       "Nunca Dudes De Mi - ['El Golpe']                                              1.079293   \n",
       "\n",
       "name_and_artists                       Turn! Turn! Turn! - ['The Byrds']  \\\n",
       "name_and_artists                                                           \n",
       "Thunderstruck - ['AC/DC']                                       1.039069   \n",
       "The Gift of Love - ['Bette Midler']                             1.030825   \n",
       "Thelma - Bonus Track - ['Paul Simon']                           1.028249   \n",
       "How I Need You - ['Bad Boys Blue']                              0.907169   \n",
       "Nunca Dudes De Mi - ['El Golpe']                                1.044738   \n",
       "\n",
       "name_and_artists                       Siéntate Ahí - [\"Oscar D'León\"]  ...  \\\n",
       "name_and_artists                                                        ...   \n",
       "Thunderstruck - ['AC/DC']                                     1.010914  ...   \n",
       "The Gift of Love - ['Bette Midler']                           1.156865  ...   \n",
       "Thelma - Bonus Track - ['Paul Simon']                         0.984139  ...   \n",
       "How I Need You - ['Bad Boys Blue']                            0.963341  ...   \n",
       "Nunca Dudes De Mi - ['El Golpe']                              1.069386  ...   \n",
       "\n",
       "name_and_artists                       I WANNA SEE SOME ASS (feat. jetsonmade) - ['Jack Harlow', 'jetsonmade']  \\\n",
       "name_and_artists                                                                                                 \n",
       "Thunderstruck - ['AC/DC']                                                       1.193503                         \n",
       "The Gift of Love - ['Bette Midler']                                             1.191703                         \n",
       "Thelma - Bonus Track - ['Paul Simon']                                           0.959309                         \n",
       "How I Need You - ['Bad Boys Blue']                                              0.894979                         \n",
       "Nunca Dudes De Mi - ['El Golpe']                                                1.041934                         \n",
       "\n",
       "name_and_artists                       MODUS - ['Joji']  Psycho! - ['MASN']  \\\n",
       "name_and_artists                                                              \n",
       "Thunderstruck - ['AC/DC']                      1.037507            1.180821   \n",
       "The Gift of Love - ['Bette Midler']            0.980848            1.029696   \n",
       "Thelma - Bonus Track - ['Paul Simon']          1.138327            0.954939   \n",
       "How I Need You - ['Bad Boys Blue']             1.087290            1.064838   \n",
       "Nunca Dudes De Mi - ['El Golpe']               1.043607            1.021067   \n",
       "\n",
       "name_and_artists                       COOLER THAN A BITCH (feat. Roddy Ricch) - ['Gunna', 'Roddy Ricch']  \\\n",
       "name_and_artists                                                                                            \n",
       "Thunderstruck - ['AC/DC']                                                       1.108988                    \n",
       "The Gift of Love - ['Bette Midler']                                             1.081106                    \n",
       "Thelma - Bonus Track - ['Paul Simon']                                           0.973634                    \n",
       "How I Need You - ['Bad Boys Blue']                                              0.982186                    \n",
       "Nunca Dudes De Mi - ['El Golpe']                                                1.060800                    \n",
       "\n",
       "name_and_artists                       What If I Told You That I Love You - ['Ali Gatie']  \\\n",
       "name_and_artists                                                                            \n",
       "Thunderstruck - ['AC/DC']                                                       1.186523    \n",
       "The Gift of Love - ['Bette Midler']                                             0.990258    \n",
       "Thelma - Bonus Track - ['Paul Simon']                                           0.906705    \n",
       "How I Need You - ['Bad Boys Blue']                                              1.123387    \n",
       "Nunca Dudes De Mi - ['El Golpe']                                                1.098322    \n",
       "\n",
       "name_and_artists                       24 (feat. Lil Baby) - ['Money Man', 'Lil Baby']  \\\n",
       "name_and_artists                                                                         \n",
       "Thunderstruck - ['AC/DC']                                                     1.097103   \n",
       "The Gift of Love - ['Bette Midler']                                           1.083273   \n",
       "Thelma - Bonus Track - ['Paul Simon']                                         1.065129   \n",
       "How I Need You - ['Bad Boys Blue']                                            1.006332   \n",
       "Nunca Dudes De Mi - ['El Golpe']                                              1.080464   \n",
       "\n",
       "name_and_artists                       Bean (Kobe) [feat. Chief Keef] - ['Lil Uzi Vert', 'Chief Keef']  \\\n",
       "name_and_artists                                                                                         \n",
       "Thunderstruck - ['AC/DC']                                                       1.048400                 \n",
       "The Gift of Love - ['Bette Midler']                                             1.216447                 \n",
       "Thelma - Bonus Track - ['Paul Simon']                                           1.137969                 \n",
       "How I Need You - ['Bad Boys Blue']                                              1.132095                 \n",
       "Nunca Dudes De Mi - ['El Golpe']                                                1.094262                 \n",
       "\n",
       "name_and_artists                       Pardon (feat. Lil Baby) - ['T.I.', 'Lil Baby']  \\\n",
       "name_and_artists                                                                        \n",
       "Thunderstruck - ['AC/DC']                                                    1.078885   \n",
       "The Gift of Love - ['Bette Midler']                                          1.199216   \n",
       "Thelma - Bonus Track - ['Paul Simon']                                        1.009152   \n",
       "How I Need You - ['Bad Boys Blue']                                           1.011595   \n",
       "Nunca Dudes De Mi - ['El Golpe']                                             1.040128   \n",
       "\n",
       "name_and_artists                       Long Live - ['Florida Georgia Line']  \\\n",
       "name_and_artists                                                              \n",
       "Thunderstruck - ['AC/DC']                                          1.027577   \n",
       "The Gift of Love - ['Bette Midler']                                1.116030   \n",
       "Thelma - Bonus Track - ['Paul Simon']                              1.163957   \n",
       "How I Need You - ['Bad Boys Blue']                                 1.011847   \n",
       "Nunca Dudes De Mi - ['El Golpe']                                   1.003542   \n",
       "\n",
       "name_and_artists                       Billetes Azules (with J Balvin) - ['KEVVO', 'J Balvin']  \n",
       "name_and_artists                                                                                \n",
       "Thunderstruck - ['AC/DC']                                                       1.103890        \n",
       "The Gift of Love - ['Bette Midler']                                             1.238159        \n",
       "Thelma - Bonus Track - ['Paul Simon']                                           1.043010        \n",
       "How I Need You - ['Bad Boys Blue']                                              0.994169        \n",
       "Nunca Dudes De Mi - ['El Golpe']                                                1.110944        \n",
       "\n",
       "[5 rows x 56798 columns]"
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "rec_df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "---\n",
    "---\n",
    "### Section 3\n",
    "#### Recommender Algorithm"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "your_song_data = pd.DataFrame()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Defining global sample rate, frame size, and hop length for incoming audio\n",
    "sr = 44100\n",
    "frame = 2048\n",
    "hop = 512"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Defining a filepath and loading one of my songs below\n",
    "rundown_wav = '../songs_test/RUNDOWN.wav'\n",
    "rundown, sr = lib.load(rundown_wav, sr=sr)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "**BELOW:** This is first working iteration of the input function that will process a new artist's track, and prepare it for comparison against the recommender dataframe above."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [],
   "source": [
    "# Defining a function to take in an audio file, perform various metric calculations on it, and return a row that can be compared to my existing dataframe\n",
    "def my_song_stats(audio_file, sr=sr, n_fft=frame, hop_length=hop, units='time'):\n",
    "    \n",
    "    # DURATION AND TEMPO METRICS\n",
    "    # Using Librosa's .get_duration() method to determine song duration in seconds\n",
    "    duration = round(lib.get_duration(audio_file, sr=sr, n_fft=frame, hop_length=hop), 1)\n",
    "    # Using Librosa's .beat_track() method to estimate tempo\n",
    "    tempo, beats = lib.beat.beat_track(audio_file, sr=sr, hop_length=hop, units=units)\n",
    "    tempo = round(tempo, 1)\n",
    "    # ---------------\n",
    "    \n",
    "    # MUSICAL KEY METRICS\n",
    "    # Using Librosa's .chroma_cqt() method to compute a chromagram vector for incoming audio\n",
    "    chroma_array = lib.feature.chroma_cqt(audio_file, sr=sr, hop_length=hop)\n",
    "    chroma_array = sp.stats.zscore(chroma_array)\n",
    "    \n",
    "    # Coefficients are defined by Krumhansl and Schmuckler, and sourced from http://davidtemperley.com/wp-content/uploads/2015/11/temperley-mp99.pdf\n",
    "    # This block references https://gist.github.com/bmcfee/1f66825cef2eb34c839b42dddbad49fd\n",
    "    major = np.asarray([6.35, 2.23, 3.48, 2.33, 4.38, 4.09, 2.52, 5.19, 2.39, 3.66, 2.29, 2.88])\n",
    "    major = sp.stats.zscore(major)\n",
    "    # Again here, coefficients are defined by Krumhansl and Schmuckler\n",
    "    minor = np.asarray([6.33, 2.68, 3.52, 5.38, 2.60, 3.53, 2.54, 4.75, 3.98, 2.69, 3.34, 3.17])\n",
    "    minor = sp.stats.zscore(minor)\n",
    "    # Using Scipy's .circulant() method to determine all major and minor key rotations\n",
    "    # Referencing https://docs.scipy.org/doc/scipy/reference/generated/scipy.linalg.circulant.html\n",
    "    major = sp.linalg.circulant(major)\n",
    "    minor = sp.linalg.circulant(minor)\n",
    " \n",
    "    # Creating empty lists to hold mean major and minor coefficients\n",
    "    maj_coeffs = []\n",
    "    min_coeffs = []\n",
    "    # Iterating through major key vectors\n",
    "    for maj_key_weights in major.T.dot(chroma_array):\n",
    "        # Taking the mean of each vector and appending to the 'maj_coeffs' list\n",
    "        maj_coeffs.append(maj_key_weights.mean())\n",
    "    # Iterating through minor key vectors\n",
    "    for min_key_weights in minor.T.dot(chroma_array):\n",
    "        # Taking the mean of each vector and appending to the 'min_coeffs' list\n",
    "        min_coeffs.append(min_key_weights.mean())\n",
    "    \n",
    "    # Creating dictionaries of keys and values for enumerated major and minor key, thereby indexing key coefficients\n",
    "    maj_keys = {index_maj: majk for (index_maj, majk) in enumerate(maj_coeffs)}\n",
    "    min_keys = {index_min: mink for (index_min, mink) in enumerate(min_coeffs)}\n",
    "    # Finding the maximum value in each dictionary along with the associated dictionary key\n",
    "    # The dictionary key also represents a letter in the chromatic scale, starting at C = 0 and increasing sequentially\n",
    "    max_maj = max(maj_keys, key=maj_keys.get)\n",
    "    max_min = max(min_keys, key=min_keys.get)\n",
    "    # Defining the best estimates for major and minor keys\n",
    "    best_maj = (max_maj, maj_keys[max_maj])[0]\n",
    "    best_min = (max_min, min_keys[max_min])[0]\n",
    "    # ---------------\n",
    "    \n",
    "    return duration, tempo, best_maj, best_min"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(216.1, 79.5, 7, 2)"
      ]
     },
     "execution_count": 30,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "my_song_stats(rundown)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}