From b6676e73617ce41c91672d33a794042618b8c877 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 10 Nov 2022 20:50:56 +0000 Subject: [PATCH] switch from sparse to normal crossentropy --- 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 f4d12d6..ce8977f 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py +++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py @@ -55,7 +55,7 @@ def model_rainfallwater_segmentation(metadata, shape_water_out, model_arch="conv model.compile( optimizer="Adam", - loss=tf.keras.losses.SparseCategoricalCrossentropy(), + loss=tf.keras.losses.CategoricalCrossentropy(), metrics=[tf.keras.metrics.SparseCategoricalAccuracy()] )