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