15 / 30 : set n_epoch to 1 if CI

Author: beachdweller

15_optimization/030_Classification_Optimization.ipynb

@@ -35,6 +35,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "import os\n",
+    "\n",
+    "\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import numpy.random as nr\n",
@@ -1002,6 +1005,12 @@
     "\n",
     "\n",
     "# Train the model\n",
+    "n_epoch = 50\n",
+    "\n",
+    "# To save time when automatically testing the SW\n",
+    "if os.getenv('CI', False):\n",
+    "    n_epoch = 1\n",
+    "\n",
     "model.fit(X, y, epochs=50, batch_size=10)\n",
     "\n",
     "\n",