The last chart you presented in Medium as prediction doesn't appear correct to me - it's still a fit to the test dataset, right?