Removed usage of CPU threads from notebooks

Author: FS-CodeBase

pain_emnist_tf.ipynb

@@ -13,7 +13,7 @@
     "**Implementation:** PAIN architecture presented in:\n",
     "**[IEEE publication](https://ieeexplore.ieee.org/abstract/document/8682767)**\n",
     "\n",
-    "***NOTE:*** *Code is setup to reconstruct 7x7 compressed noisy images. Training parameters may need to be adjusted for 4x4, 14x14, and 28x28 cases.*\n",
+    "***NOTE:*** *Code is setup to reconstruct 7x7 compressed noisy images. Batch size and learning rates may need to be adjusted for 4x4, 14x14, and 28x28 cases.*\n",
     "\n",
     "Python Version: 3.11.5  \n",
     "Jupyer Notebook Version: 6.5.4\n",
@@ -231,15 +231,6 @@
     "clean_test  = np.array([matrix.reshape(out_dim**2,) for matrix in clean_test/255])"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "[[clean_test.shape, clean_train.shape],[noisy_test.shape, noisy_train.shape]]"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -301,9 +292,6 @@
    },
    "outputs": [],
    "source": [
-    "# Set threads using 6 CPU cores\n",
-    "tf.config.threading.set_intra_op_parallelism_threads(6)\n",
-    "\n",
     "# Create & Compile the PAIN model\n",
     "PAIN = build_PAIN(in_dim=cmp_dim, out_dim=out_dim)\n",
     "\n",

pain_tf.ipynb

@@ -13,7 +13,7 @@
     "**Implementation:** PAIN architecture presented in:\n",
     "**[IEEE publication](https://ieeexplore.ieee.org/abstract/document/8682767)**\n",
     "\n",
-    "***NOTE:*** Code is setup to reconstruct 7x7 compressed noisy images. Training parameters may need to be adjusted for 4x4, 14x14, and 28x28 cases.\n",
+    "***NOTE:*** *Code is setup to reconstruct 7x7 compressed noisy images. Batch size and learning rates may need to be adjusted for 4x4, 14x14, and 28x28 cases.*\n",
     "\n",
     "Python Version: 3.11.5  \n",
     "Jupyer Notebook Version: 6.5.4"
@@ -35,11 +35,14 @@
     "# Import necessary modules and libraries \n",
     "from __future__ import division, print_function, absolute_import\n",
     "import tensorflow as tf\n",
+    "from tensorflow.keras.models import Model\n",
     "import numpy as np\n",
     "import matplotlib.pyplot as plt\n",
     "import h5py\n",
     "import os\n",
-    "from skimage.util.shape import view_as_windows"
+    "from skimage.util.shape import view_as_windows\n",
+    "from sklearn.manifold import TSNE\n",
+    "from sklearn.decomposition import PCA"
    ]
   },
   {
@@ -202,7 +205,7 @@
     "dat_file = f'{directory}/mnist_{cmp_dim}x{cmp_dim}_train.h5'\n",
     "    \n",
     "# Load MNIST\n",
-    "(clean_train, _), (clean_test, _) = tf.keras.datasets.mnist.load_data()\n",
+    "(clean_train, train_labels), (clean_test, test_labels) = tf.keras.datasets.mnist.load_data()\n",
     "\n",
     "# If the compressed/noisy data does not exist, create it\n",
     "if not os.path.exists(dat_file):\n",
@@ -288,9 +291,6 @@
    },
    "outputs": [],
    "source": [
-    "# Set threads using 6 CPU cores\n",
-    "tf.config.threading.set_intra_op_parallelism_threads(6)\n",
-    "\n",
     "# Create & Compile the PAIN model\n",
     "PAIN = build_PAIN(in_dim=cmp_dim, out_dim=out_dim)\n",
     "\n",

pics_tf.ipynb

@@ -13,7 +13,7 @@
     "**Implementation:** PICS architecture discussed in:\n",
     "**[IEEE publication](https://ieeexplore.ieee.org/abstract/document/8682767)**\n",
     "\n",
-    "***NOTE:*** *Code is setup to reconstruct 7x7 compressed noisy images. Training parameters may need to be adjusted for 4x4, 14x14, and 28x28 cases.*\n",
+    "***NOTE:*** *Code is setup to reconstruct 7x7 compressed noisy images. Batch size and learning rates may need to be adjusted for 4x4, 14x14, and 28x28 cases.*\n",
     "\n",
     "Python Version: 3.11.5  \n",
     "Jupyer Notebook Version: 6.5.4"
@@ -304,9 +304,6 @@
    },
    "outputs": [],
    "source": [
-    "# Set threads using 6 CPU cores\n",
-    "tf.config.threading.set_intra_op_parallelism_threads(6)\n",
-    "\n",
     "# Create & Compile the PICS model\n",
     "PICS = build_PICS(in_dim=cmp_dim, out_dim=out_dim, flt_dim=2)\n",
     "\n",
@@ -370,9 +367,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Print examples: training set\n",
-    "################################\n",
-    "\n",
     "# Create a 3 by 4 subplot handle\n",
     "fig, axes = plt.subplots(3, 4, figsize=(9, 8))\n",
     "axes      = axes.flatten()\n",
@@ -439,9 +433,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# Print examples: test set\n",
-    "################################\n",
-    "\n",
     "# Create a 3 by 4 subplot handle\n",
     "fig, axes = plt.subplots(3, 4, figsize=(9, 8))\n",
     "axes      = axes.flatten()\n",