Необходимо предсказывать временные ряды

Нужно предсказывать временные ряды. Временные ряды представляют из себя набор объектов с фиксированным количеством точек, каждая точка имеет 2d координаты. В каждом последующем временном ряде может быть разное количество объектов. Также необходимо, чтобы модель находила зависимости одного объекта от другого и в результате выдавала не конкретное предсказание временного ряда, а список вероятных предсказаний. Есть модель от amazon chronos для временных рядов, но она вроде для каждого объекта предсказывает отдельно без зависимостей. Пример: item_id timestamp x1 y1 x2 y2 x3 y3 1 100 7 12 2 4 3 11 4 100 3 14 5 8 4 1 3 100 1 23 4 12 1 9 1 101 8 13 3 5 4 12