Create image directory and check for sklearn >= 0.20

02_end_to_end_machine_learning_project.ipynb

@@ -22,7 +22,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "First, let's import a few common modules, ensure MatplotLib plots figures inline and prepare a function to save the figures. We also check that Python 3.5 or later is installed (although Python 2.x may work, it is deprecated so we strongly recommend you use Python 3 instead)."
+    "First, let's import a few common modules, ensure MatplotLib plots figures inline and prepare a function to save the figures. We also check that Python 3.5 or later is installed (although Python 2.x may work, it is deprecated so we strongly recommend you use Python 3 instead), as well as Scikit-Learn ≥0.20."
    ]
   },
   {
@@ -35,6 +35,10 @@
     "import sys\n",
     "assert sys.version_info >= (3, 5)\n",
     "\n",
+    "# Scikit-Learn ≥0.20 is required\n",
+    "import sklearn\n",
+    "assert sklearn.__version__ >= \"0.20\"\n",
+    "\n",
     "# Common imports\n",
     "import numpy as np\n",
     "import os\n",
@@ -54,6 +58,7 @@
     "PROJECT_ROOT_DIR = \".\"\n",
     "CHAPTER_ID = \"end_to_end_project\"\n",
     "IMAGES_PATH = os.path.join(PROJECT_ROOT_DIR, \"images\", CHAPTER_ID)\n",
+    "os.makedirs(IMAGES_PATH, exist_ok=True)\n",
     "\n",
     "def save_fig(fig_id, tight_layout=True, fig_extension=\"png\", resolution=300):\n",
     "    path = os.path.join(IMAGES_PATH, fig_id + \".\" + fig_extension)\n",
@@ -67,23 +72,6 @@
     "warnings.filterwarnings(action=\"ignore\", message=\"^internal gelsd\")"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "This notebook assumes you have installed Scikit-Learn ≥0.20."
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 2,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "import sklearn\n",
-    "assert sklearn.__version__ >= \"0.20\""
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},