Выбор точной модели времени бега

тема: Метрики регрессии и МНК · уровень: средний

Условие

В дневнике тренировок бегуна для нескольких пробежек сохранено фактическое время и прогнозы двух моделей: SprintNet и PacePro. Время измеряется в секундах.

У некоторых пробежек прогноз одной из моделей может отсутствовать и обозначается символом -. Для такой модели эта пробежка не участвует в расчёте ошибки. Фактическое время присутствует в каждой строке.

Для каждой модели требуется вычислить RMSE, среднеквадратичную ошибку прогноза:

RMSE = sqrt(((y1 - p1)^2 + (y2 - p2)^2 + ... + (yk - pk)^2) / k),

где y — фактическое время, p — прогноз модели, а k — число строк, в которых у этой модели есть прогноз. У каждой модели есть хотя бы один прогноз, поэтому деления на ноль не возникает.

Требуется вывести имя модели с меньшим RMSE. Округление не применяется: требуется вывести только имя модели. Если RMSE моделей совпадают, требуется вывести PacePro.

Формат ввода

В первой строке дано целое число n — количество пробежек.

В следующих n строках даны три значения: фактическое время пробежки, прогноз SprintNet, прогноз PacePro. Прогноз может быть целым числом или символом -.

Формат вывода

Выведите SprintNet, если её RMSE меньше.

Иначе выведите PacePro.

Ограничения

Решить задачу с автопроверкой на Python →

Куда дальше