sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-22 09:13:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

LossCrossentropy: don't sum

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-12-08 16:57:11 +00:00
parent 08046340f4
commit a3c9416cf0
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
aimodel/src/lib/ai/components/LossCrossentropy.py
View file

@ -16,10 +16,10 @@ class LossCrossentropy(tf.keras.losses.Loss):
def call(self, y_true, y_pred): def call(self, y_true, y_pred):
result = tf.keras.metrics.categorical_crossentropy(y_true, y_pred) result = tf.keras.metrics.categorical_crossentropy(y_true, y_pred)
result_reduce = tf.math.reduce_sum(result) # result_reduce = tf.math.reduce_sum(result)
label_nowater = tf.math.reduce_sum(tf.argmax(y_true, axis=-1)) # label_nowater = tf.math.reduce_sum(tf.argmax(y_true, axis=-1))
# tf.print("DEBUG:TFPRINT:loss LABEL", y_true.shape, y_true, "LABEL_ARGMAX_COUNT_AXIS0", label_nowater, "PREDICT", y_pred.shape, y_pred, "BEFORE_REDUCE", result.shape, result, "AFTER_REDUCE", result_reduce.shape, result_reduce) # tf.print("DEBUG:TFPRINT:loss LABEL", y_true.shape, y_true, "LABEL_ARGMAX_COUNT_AXIS0", label_nowater, "PREDICT", y_pred.shape, y_pred, "BEFORE_REDUCE", result.shape, result, "AFTER_REDUCE", result_reduce.shape, result_reduce)
return result_reduce return result
def get_config(self): def get_config(self):
config = super(LossCrossentropy, self).get_config() config = super(LossCrossentropy, self).get_config()