Thanks, that did the trick. Any idea what the output of ConfusionMatrix means for num_classes=1?
The documentation is thin on this as well, and it returns a 1x1 tensor / single value (I expected a 2x2 matrix). I can’t make sense of it based on my precision/recall.