From 030d8710b63955b7c69085ed6ec5cc6825861dda Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 28 Sep 2022 18:08:31 +0100 Subject: [PATCH] fixup --- aimodel/src/lib/ai/model_rainfallwater_segmentation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py index 126fd73..87b4bab 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py +++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py @@ -31,7 +31,7 @@ def model_rainfallwater_segmentation(metadata, feature_dim_in, shape_water_out, model.compile( optimizer="Adam", - loss=tf.keras.losses.SparseCategoricalCrossentropy() + loss=tf.keras.losses.SparseCategoricalCrossentropy(), metrics=["accuracy"] )