Класс RobotModel

Класс представляющий модель робота.

Обязательные поля

model - нейросетевая модель (dtype torch.nn.Module или класс наследник) например может быть использована DefaultModel

Обязательные методы

get_optimizer - создает и возвращает оптимайзер (torch.optim)
get_loss_fn - создает и возвращает loss функцию
update_state - функция перехода робота в следующее состояние
forward - функция, которая вызывается при исползовании модели (расчет при вызове)
calc_metrics - расчет дополнительных метрик

Опционально

plot_trajectories - постоение графиков траекторий
save_predict_to_csv - сохранение данных в формает csv