I have a ConfusionMatrix metric implemented as a subclass of tf.keras.metrics.Metric where the only value is a tf.Variable with shape=(13,13) and dtype=tf.uint32. In wandb UI this just shows up as a histogram, is there some way to reinterpret this as a multilabel confusion matrix?

Hey Tom, it’s not possible at the moment. You’ll need to use wandb.plot.confusion_matrix() for that.