mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-25 18:33:01 +00:00
almost got it....? it's not what I expected....!
This commit is contained in:
parent
f48473b703
commit
6640a41bb7
1 changed files with 1 additions and 0 deletions
|
@ -17,6 +17,7 @@ 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))
|
||||||
tf.print("DEBUG:TFPRINT:loss LABEL", y_true.shape, y_true, "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, "PREDICT", y_pred.shape, y_pred, "BEFORE_REDUCE", result.shape, result, "AFTER_REDUCE", result_reduce.shape, result_reduce)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue