I added a little code to include a vertical dashed line on the plot representing where total_weights = number of train observations. I also moved n_epochs as an argument to fit_model() so learners can play around with the impact of n_epochs more easily.
66 KiB
66 KiB