From f37284bf83fea43e0540fa056f86f80bed48e598 Mon Sep 17 00:00:00 2001
From: Lilly Thomas <lilly@developmentseed.org>
Date: Fri, 12 Jan 2024 21:35:46 -0800
Subject: [PATCH] rename notebook, add early stopping for clay finetuned
 variant

---
 notebooks/flood_benchmark_segmentation.ipynb | 58 ++------------------
 1 file changed, 4 insertions(+), 54 deletions(-)

diff --git a/notebooks/flood_benchmark_segmentation.ipynb b/notebooks/flood_benchmark_segmentation.ipynb
index 4dafe6c9..4d4f61f6 100644
--- a/notebooks/flood_benchmark_segmentation.ipynb
+++ b/notebooks/flood_benchmark_segmentation.ipynb
@@ -3119,7 +3119,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 70,
+      "execution_count": null,
       "id": "8jqeBd6LOLkr",
       "metadata": {
         "colab": {
@@ -3128,32 +3128,7 @@
         "id": "8jqeBd6LOLkr",
         "outputId": "36bcccb7-ef5c-4c78-e9b8-c02eaf85ea0a"
       },
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "IoU: 0.3379289507865906\n",
-            "IoU: 0.5677745938301086\n",
-            "IoU: 0.38128602504730225\n",
-            "IoU: 0.4337167739868164\n",
-            "IoU: 0.07514558732509613\n",
-            "IoU: 0.8329333066940308\n",
-            "IoU: 0.16996563971042633\n",
-            "IoU: 0.2993166744709015\n",
-            "IoU: 0.8047876954078674\n",
-            "IoU: 0.36587828397750854\n",
-            "IoU: 0.9110873341560364\n",
-            "IoU: 0.015298277139663696\n",
-            "IoU: 0.8980720043182373\n",
-            "IoU: 0.08573030680418015\n",
-            "IoU: 0.4487551748752594\n",
-            "IoU: 0.8745391964912415\n",
-            "IoU: 0.18230441212654114\n",
-            "Average IoU: 0.45203059911727905\n"
-          ]
-        }
-      ],
+      "outputs": [],
       "source": [
         "IoUs_fs = calculate_IoU(det_model, dm.val_dataloader(), clay=False)\n",
         "\n",
@@ -3164,7 +3139,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 71,
+      "execution_count": null,
       "id": "6FPXD6tmKdB2",
       "metadata": {
         "colab": {
@@ -3173,32 +3148,7 @@
         "id": "6FPXD6tmKdB2",
         "outputId": "bcd5d8cd-b66f-44c2-dea4-bf75dfe6b8ee"
       },
-      "outputs": [
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "IoU: 0.0901893898844719\n",
-            "IoU: 0.4645256996154785\n",
-            "IoU: 0.3826591372489929\n",
-            "IoU: 0.0009378832764923573\n",
-            "IoU: 0.06925993412733078\n",
-            "IoU: 0.6750580072402954\n",
-            "IoU: 0.32923126220703125\n",
-            "IoU: 0.2249605357646942\n",
-            "IoU: 0.8142861723899841\n",
-            "IoU: 0.3637990951538086\n",
-            "IoU: 0.28369003534317017\n",
-            "IoU: 0.056073836982250214\n",
-            "IoU: 0.778351366519928\n",
-            "IoU: 0.20255421102046967\n",
-            "IoU: 0.2703555226325989\n",
-            "IoU: 0.7836553454399109\n",
-            "IoU: 0.08511418849229813\n",
-            "Average IoU from model finetuned from Clay: 0.3455706834793091\n"
-          ]
-        }
-      ],
+      "outputs": [],
       "source": [
         "IoUs_cl = calculate_IoU(det_model_clay, dm.val_dataloader(), clay=True)\n",
         "\n",