stash.diff

diff --git a/tsaplay/utils/addons.py b/tsaplay/utils/addons.py
index fef0c92..4203b11 100644
--- a/tsaplay/utils/addons.py
+++ b/tsaplay/utils/addons.py
@@ -146,7 +146,7 @@ def conf_matrix(model, features, labels, spec, params):
                     depth=params["_n_out_classes"],
                 ),
                 num_classes=params["_n_out_classes"],
-            )
+            ),
         }
     )
     eval_hooks += [
@@ -217,7 +217,26 @@ def scalars(model, features, labels, spec, params):
         ),
         "auc": tf.metrics.auc(
             labels=tf.one_hot(indices=labels, depth=params["_n_out_classes"]),
-            predictions=spec.predictions["probabilities"],
+            predictions=tf.Print(
+                input_=spec.predictions["probabilities"],
+                data=[
+                    spec.predictions["probabilities"],
+                    tf.concat(
+                        [
+                            features["left"],
+                            features["target"],
+                            features["right"],
+                        ],
+                        axis=1,
+                    ),
+                    # features["left"],
+                    tf.expand_dims(features["target"], axis=2),
+                    # features["right"],
+                    labels,
+                ],
+                message="",
+                summarize=5000,
+            ),
             name="auc_op",
         ),
     }