From 7d7ccce9d367f18360c1467ba659c5ea7007494f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= <ageron@users.noreply.github.com>
Date: Fri, 20 Oct 2017 20:05:09 +0200
Subject: [PATCH] Fix factorize() bug, due to housing_cat being a DataFrame,
 not a Series, fixes #104

---
 02_end_to_end_machine_learning_project.ipynb | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/02_end_to_end_machine_learning_project.ipynb b/02_end_to_end_machine_learning_project.ipynb
index 4efc429..7a72959 100644
--- a/02_end_to_end_machine_learning_project.ipynb
+++ b/02_end_to_end_machine_learning_project.ipynb
@@ -667,7 +667,8 @@
    },
    "outputs": [],
    "source": [
-    "housing_num = housing.select_dtypes(include=[np.number])"
+    "housing_num = housing.drop('ocean_proximity', axis=1)\n",
+    "# alternatively: housing_num = housing.select_dtypes(include=[np.number])"
    ]
   },
   {
@@ -775,7 +776,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "housing_cat = housing.select_dtypes(exclude=[np.number])\n",
+    "housing_cat = housing['ocean_proximity']\n",
     "housing_cat.head(10)"
    ]
   },