From 94fd5938653544648b174b4e68f9ac3c3d88ba35 Mon Sep 17 00:00:00 2001 From: root Date: Sun, 29 May 2022 21:47:20 -0600 Subject: [PATCH] use tf keras --- notebooks/wut-train.ipynb | 29 ++++++++++------------------- 1 file changed, 10 insertions(+), 19 deletions(-) diff --git a/notebooks/wut-train.ipynb b/notebooks/wut-train.ipynb index e8fb2be..739f460 100644 --- a/notebooks/wut-train.ipynb +++ b/notebooks/wut-train.ipynb @@ -45,20 +45,20 @@ "import tensorflow as tf\n", "from tensorflow import keras\n", "from tensorflow.keras import layers\n", - "\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.preprocessing.image import ImageDataGenerator\n", "from tensorflow.keras.layers import Convolution2D, MaxPooling2D, ZeroPadding2D\n", - "from tensorflow.keras import optimizers\n", - "from tensorflow.keras.preprocessing import image\n", - "from tensorflow.keras.models import load_model\n", - "from tensorflow.keras.preprocessing.image import load_img\n", - "from tensorflow.keras.preprocessing.image import img_to_array\n", "from tensorflow.keras.layers import Dense, Conv2D, Flatten, Dropout, MaxPooling2D\n", - "from tensorflow.keras.models import Model\n", "from tensorflow.keras.layers import Input, concatenate\n", - "from tensorflow.keras.utils import model_to_dot" + "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" ] }, { @@ -300,7 +300,7 @@ "outputs": [], "source": [ "#wutoptimizer = 'adam'\n", - "wutoptimizer = keras.optimizers.Adam(learning_rate=0.001, beta_1=0.9, beta_2=0.999, amsgrad=True)\n", + "wutoptimizer = tf.keras.optimizers.Adam(learning_rate=0.001, beta_1=0.9, beta_2=0.999, amsgrad=True)\n", "\n", "wutloss = 'binary_crossentropy'\n", "#wutmetrics = 'accuracy'\n", @@ -457,15 +457,6 @@ "model.save('/srv/satnogs/data/models/GMSK/wut-GMSK-202205.tf')" ] }, - { - "cell_type": "code", - "execution_count": null, - "metadata": {}, - "outputs": [], - "source": [ - "from keras.utils import plot_model" - ] - }, { "cell_type": "code", "execution_count": null,