Create a confusion matrix in tensorboard Track training and validation scores of your model for every epoch. Installation pip install -e git+https://github.com/mr-francois/cmtb.git#egg=cmtb Example See this notebook