diff --git a/choice_learn/tf_ops.py b/choice_learn/tf_ops.py index de39ac1d..3981a273 100644 --- a/choice_learn/tf_ops.py +++ b/choice_learn/tf_ops.py @@ -105,6 +105,7 @@ def __init__( reduction: Reduction function - here to follow tf.keras.losses.Loss signature """ + print("Reduction", reduction) super().__init__(reduction=reduction, name=name) self.label_smoothing = label_smoothing self.from_logits = from_logits