Create a confusion matrix in tensorboard Track training and validation scores of your model for every epoch. Installation pip install cmtb Example See this notebook