From 32322b7f1569ab5d4256aa9a097a35bf4c61b57c Mon Sep 17 00:00:00 2001
From: root <root@localhost>
Date: Sun, 29 May 2022 21:51:06 -0600
Subject: [PATCH] use tf keras

---
 notebooks/wut.ipynb | 47 +++++++++++++++++++++++++--------------------
 1 file changed, 26 insertions(+), 21 deletions(-)

diff --git a/notebooks/wut.ipynb b/notebooks/wut.ipynb
index 845f1fb..abee6f8 100644
--- a/notebooks/wut.ipynb
+++ b/notebooks/wut.ipynb
@@ -63,6 +63,15 @@
     "import os"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -78,27 +87,23 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import keras"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from keras import Sequential\n",
-    "from keras.layers import Activation, Dropout, Flatten, Dense\n",
-    "from keras.preprocessing.image import ImageDataGenerator\n",
-    "from keras.layers import Convolution2D, MaxPooling2D, ZeroPadding2D\n",
-    "from keras import optimizers\n",
-    "from keras.preprocessing import image\n",
-    "from keras.models import load_model\n",
-    "#from keras.preprocessing.image import load_img\n",
-    "#from keras.preprocessing.image import img_to_array\n",
-    "from keras.layers import Dense, Conv2D, Flatten, Dropout, MaxPooling2D\n",
-    "from keras.models import Model\n",
-    "from keras.layers import Input, concatenate"
+    "import tensorflow as tf\n",
+    "from tensorflow import keras\n",
+    "from tensorflow.keras import layers\n",
+    "from tensorflow.keras import optimizers\n",
+    "from tensorflow.keras import Sequential\n",
+    "from tensorflow.keras.layers import Activation, Dropout, Flatten, Dense\n",
+    "from tensorflow.keras.layers import Convolution2D, MaxPooling2D, ZeroPadding2D\n",
+    "from tensorflow.keras.layers import Dense, Conv2D, Flatten, Dropout, MaxPooling2D\n",
+    "from tensorflow.keras.layers import Input, concatenate\n",
+    "from tensorflow.keras.models import load_model\n",
+    "from tensorflow.keras.models import Model\n",
+    "from tensorflow.keras.preprocessing import image\n",
+    "from tensorflow.keras.preprocessing.image import ImageDataGenerator\n",
+    "from tensorflow.keras.preprocessing.image import img_to_array\n",
+    "from tensorflow.keras.preprocessing.image import load_img\n",
+    "from tensorflow.keras.utils import model_to_dot\n",
+    "from tensorflow.keras.utils import plot_model"
    ]
   },
   {