This commit is contained in:
Starbeamrainbowlabs 2022-09-28 18:08:31 +01:00
parent 4ee7f2a0d6
commit 030d8710b6
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -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"]
)