K-Fold Cross-Validation from scratch

MediumPython
05:00
Лучше работает на десктопе
Applied DScross-validationk-foldMAE

Реализуйте K-Fold Cross-Validation.

Для каждого фолда:
1. Используйте его как тестовый набор, остальные — тренировочный
2. Модель = среднее тренировочной выборки (как предсказание)
3. Вычислите MAE на тестовом фолде

Сигнатура

def kfold_cv(values: list[float], k: int) -> dict:

Верните dict:
- fold_maes: список MAE каждого фолда (округлить до 4 знаков)
- mean_mae: среднее MAE (округлить до 4 знаков)

Примеры

kfold_cv([1,2,3,4,5,6], 3) → {"fold_maes": [3.0, 0.5, 3.0], "mean_mae": 2.1667}

Примеры

Пример 1

Вход:
values = [1,2,3,4,5,6]
k = 3
Выход:{"fold_maes":[3,0.5,3],"mean_mae":2.1667}

Пример 2

Вход:
values = [10,20,30,40]
k = 2
Выход:{"fold_maes":[20,20],"mean_mae":20}

Пример 3

Вход:
values = [1,1,1,1,1]
k = 5
Выход:{"fold_maes":[0,0,0,0,0],"mean_mae":0}
Консоль
Нажмите Run или Ctrl+Enter для запуска