how to plot confusion_matrix with lightning tensorboard?

[jaffe-fly]

I need to plot confusion_matrix in tensotboard ,how to do it ?

[tshu-w]

https://stackoverflow.com/questions/65498782/how-to-dump-confusion-matrix-using-tensorboard-logger-in-pytorch-lightning

[Borda]

confusion_matrix = torchmetrics.functional.confusion_matrix(preds, targets, num_classes=10)
df_cm = pd.DataFrame(confusion_matrix.numpy(), index = range(10), columns=range(10))
fig, ax = plt.subplots(figsize = (10,7))
sns.heatmap(df_cm, ax=ax, annot=True, cmap='Spectral')
self.logger.experiment.add_figure("Confusion matrix", fig, self.current_epoch)