Currently, `ae.plot()` adds `r2` scores to the plot. Allow user to turn this off or replace with other metrics.