test loss function

Arno1235 · Mar 22, 2023 · 136267e · 136267e
1 parent f1ae79d
commit 136267e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/_weight.py b/_weight.py
@@ -9,4 +9,4 @@ def feedback_weight_map(flat_probs, flat_labels, beta, op):
     '''
     probs = tf.reduce_sum(flat_probs*flat_labels, axis=-1)
     weight_map = tf.exp(-tf.pow(probs, beta)*tf.math.log(tf.constant(op, "float")))   
-    return weight_map 
+    return weight_map