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