From 8195318a4231d354bc7bc46a8bae1ccf5cb054d9 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 21 Oct 2022 16:51:20 +0100 Subject: [PATCH] =?UTF-8?q?SparseCategoricalAccuracy:=20=20losses=20?= =?UTF-8?q?=E2=86=92=20metrics?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- 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 551d992..0330ee7 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(), - metrics=[tf.keras.losses.SparseCategoricalAccuracy()] + metrics=[tf.keras.metrics.SparseCategoricalAccuracy()] ) return model \ No newline at end of file