Added tensorboard

KD-3 · Aug 3, 2020 · c632dac · c632dac
1 parent 86de959
commit c632dac
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/BTP_Final_model_1.ipynb b/BTP_Final_model_1.ipynb
@@ -6,7 +6,7 @@
       "name": "BTP_Final_model_1",
       "provenance": [],
       "mount_file_id": "1BE7Pf9phr7c6uT5F8eS5O58z6_cCyAWc",
-      "authorship_tag": "ABX9TyN5GyVq51tIK3svlzzR+tP+"
+      "authorship_tag": "ABX9TyOvrVVydFVv+gAPNeFXh8Bn"
     },
     "kernelspec": {
       "name": "python3",
@@ -346,12 +346,18 @@
         "outputId": "dbc5fa3e-9848-4c99-ea56-31ef29ed078e"
       },
       "source": [
+        "# Load the TensorBoard notebook extension\n",
+        "%load_ext tensorboard\n",
+        "log_dir = \"logs/fit/\" + datetime.datetime.now().strftime(\"%Y%m%d-%H%M%S\")\n",
+        "tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1)\n",
+        "\n",
         "history = model.fit(\n",
         "    train_generator,\n",
         "    steps_per_epoch=8,\n",
         "    epochs=15,\n",
         "    verbose=1,\n",
-        ")"
+        ")\n",
+        "%tensorboard --logdir logs/fit\n"
       ],
       "execution_count": null,
       "outputs": [