TF2_Confusion_Matrix_Callback

Callback for Tensorflow

confusion_matrix = Add_Confusion_Matrix(log_dir=tensorboard_output, validation_data=validation_generator.data_set,
                                        validation_steps=len(validation_generator), class_names=class_names, frequency=5)

Output in Tensorboard

IGCT2021 - BMA presentation