mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-22 17:23:01 +00:00
mono: drop the sparse
This commit is contained in:
parent
7e8f63f8ba
commit
3a0356929c
1 changed files with 1 additions and 1 deletions
|
@ -72,7 +72,7 @@ def model_rainfallwater_mono(metadata, shape_water_out, model_arch_enc="convnext
|
||||||
model.compile(
|
model.compile(
|
||||||
optimizer="Adam",
|
optimizer="Adam",
|
||||||
loss=tf.keras.losses.CategoricalCrossentropy(),
|
loss=tf.keras.losses.CategoricalCrossentropy(),
|
||||||
metrics=[tf.keras.metrics.SparseCategoricalAccuracy()]
|
metrics=[tf.keras.metrics.CategoricalAccuracy()]
|
||||||
)
|
)
|
||||||
|
|
||||||
return model
|
return model
|
Loading…
Reference in a new issue