Добрый день!
Я с таким и подобными кодами работал, причем писал их сам. Данная версия построит вам практически идентичный график предсказания но в полевых условиях работать не будет. Дело в том что в таком формате модели очень быстро обучаются выдавать не следующий за периодом который мы подали отрезок, а последний из поданного периода, поскольку разница между ними не большая и соответственно схождение хорошее. Этому способствует любой банч сайз больше двух. Но если подать банч сайз один то модель перестает сходиться вовсе. Я решил в свое время эту проблему превратив ответы в бинарный вариант.