Trainer

Класс, содержащий методы для обучения нейросетевой динамиечской модели. Данный класс, не является пользовательским и не подлежит изменению

Схема взаимодействия класса с Trainer с классами RobotDataset и RobotModel, предсавлена на рисунке

Trainer взаимодействует с классом, представляющим модель робота (RosbotModel) и наборами объектов класса датасета (RosbotDataset)

Основные метода класса

fit - непосредственное обучение нейросети
evaluate - проверки нейросети
calc_loss_and_metrics - рассчет метрик и функции потерь (вызываются методы из класса RobotModel)
sample_train_batch - сэмплирование обучающего подвыборки (пачки)
predict_multi_step - предсказание траектории, на несколько шагов вперед