add methods for viewing coefficients, losses, etc. eventually incorporate plotly for animation/interaction