adding a note about scaling

pull/34/head
Jen Looper 4 years ago
parent 8f1770f5af
commit 2451ec4db8

@ -303,16 +303,21 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"from sklearn.preprocessing import LabelEncoder\n", "from sklearn.preprocessing import LabelEncoder, StandardScaler\n",
"le = LabelEncoder()\n", "le = LabelEncoder()\n",
"\n", "\n",
"# scaler = StandardScaler()\n",
"\n",
"X = df.loc[:, ('artist_top_genre','popularity','danceability','acousticness','loudness','energy')]\n", "X = df.loc[:, ('artist_top_genre','popularity','danceability','acousticness','loudness','energy')]\n",
"\n", "\n",
"y = df['artist_top_genre']\n", "y = df['artist_top_genre']\n",
"\n", "\n",
"X['artist_top_genre'] = le.fit_transform(X['artist_top_genre'])\n", "X['artist_top_genre'] = le.fit_transform(X['artist_top_genre'])\n",
"\n", "\n",
"y = le.transform(y)\n" "# X = scaler.fit_transform(X)\n",
"\n",
"y = le.transform(y)\n",
"\n"
] ]
}, },
{ {

Loading…
Cancel
Save