Try using https://proplot.readthedocs.io/en/latest/index.html for a simpler plotting experience and create different graphs with various stats based on the template training example