Сравнение прогнозов выработки по MSE

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

Условие

Для нескольких моментов времени известна фактическая выработка солнечной электростанции, прогноз базовой модели и прогноз новой модели. Часть измерений может отсутствовать и обозначается строкой NA.

Для каждой строки, где известны все три значения, необходимо вычислить квадраты ошибок базовой и новой моделей. Строки, содержащие хотя бы одно значение NA, полностью исключаются из расчёта.

Средняя квадратичная ошибка называется MSE и вычисляется по формуле MSE = (1 / k) * Σ(y_i - p_i)^2, где k — число строк без пропусков, y_i — фактическая выработка, p_i — прогноз модели. Требуется найти, во сколько раз новая модель лучше базовой по MSE: MSE_базовой / MSE_новой.

Если MSE базовой и новой моделей равны, отношение считается равным 1.00. Гарантируется, что после исключения пропусков существует хотя бы одна строка и обе MSE строго положительны.

Формат ввода

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

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

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

Выведите одно число — отношение MSE_базовой / MSE_новой с двумя знаками после точки.

Ограничения

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

Куда дальше