Chapter 2: Generic way of selecting numerical or categorical attributes

2017-10-15 04:01:07 -07:00
parent d736811a9c
commit 887e26216d
1 changed files with 2 additions and 2 deletions
--- a/02_end_to_end_machine_learning_project.ipynb
+++ b/02_end_to_end_machine_learning_project.ipynb
@@ -2247,7 +2247,7 @@
   },
   "outputs": [],
   "source": [
-    "housing_num = housing.drop(\"ocean_proximity\", axis=1)"
+    "housing_num = housing.select_dtypes(include=[np.number])"
   ]
  },
  {
@@ -2646,7 +2646,7 @@
    }
   ],
   "source": [
-    "housing_cat = housing[\"ocean_proximity\"]\n",
+    "housing_cat = housing.select_dtypes(exclude=[np.number])\n",
    "housing_cat.head(10)"
   ]
  },