Get rid of os module

Author: abdullahkady

main.ipynb

@@ -7,8 +7,7 @@
    "outputs": [],
    "source": [
     "import matplotlib.pyplot as plt\n",
-    "import numpy as np\n",
-    "import os"
+    "import numpy as np"
    ]
   },
   {
@@ -29,15 +28,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "def get_x(folder_name):\n",
-    "    sorted_file_names = sorted([x for x in os.listdir(folder_name) if x.endswith('.jpg')], key=lambda x: int(x[:-4]))\n",
-    "\n",
-    "    image_paths = [os.path.join(folder_name, x) for x in sorted_file_names]\n",
+    "def get_x(folder_name, number_images):\n",
     "    x_input_points = np.zeros((0, 784))\n",
-    "    for img in image_paths:\n",
-    "        x_input_points = np.append(x_input_points, plt.imread(img).reshape(1, 784), axis=0)\n",
+    "    for i in range(1, number_images + 1): # +1 Since it's exclusive\n",
+    "        img_path = '{}/{}.jpg'.format(folder_name, i)\n",
+    "        x_input_points = np.append(x_input_points, plt.imread(img_path).reshape(1, 784), axis=0)\n",
     "\n",
-    "    return x_input_points "
+    "    return x_input_points"
    ]
   },
   {
@@ -104,9 +101,9 @@
    "outputs": [],
    "source": [
     "# Loading in a separate cell to avoid multiple loads.\n",
-    "x_train = get_x('Train')\n",
+    "x_train = get_x('Train', 2400)\n",
     "x_delta = get_x_delta(x_train)\n",
-    "training_true_labels = get_t(os.path.join('Train', 'Training Labels.txt'))"
+    "training_true_labels = get_t('Train/Training Labels.txt')"
    ]
   },
   {
@@ -129,8 +126,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "x_delta_test = get_x_delta(get_x('Test'))\n",
-    "test_true_labels = get_t(os.path.join('Test', 'Test Labels.txt'))\n",
+    "x_delta_test = get_x_delta(get_x('Test', 200))\n",
+    "test_true_labels = get_t('Test/Test Labels.txt')\n",
     "conf_matrix = get_confusion_matrix(x_delta_test, weights_matrix, test_true_labels)\n",
     "plt.imshow(conf_matrix)\n",
     "plt.savefig('Confusion.jpg')\n",