1. Learn
    2. /
  2. Курси
    3. /
  3. Як перемагати в змаганнях Kaggle за допомогою Python
Connected

Вправа

Середнє кодування за цільовою змінною (mean target encoding)

Спершу ви створите функцію, що реалізує середнє кодування за цільовою змінною. Пам'ятайте, що потрібно виконати два кроки:

  1. Обчислити середнє на тренувальних даних і застосувати до тестових
  2. Розбити тренувальні дані на K фолдів. Для кожного фолду обчислити out-of-fold середнє та застосувати до цього конкретного фолду

Кожен із цих кроків буде реалізовано в окремій функції: відповідно test_mean_target_encoding() і train_mean_target_encoding().

Підсумкова функція mean_target_encoding() приймає як аргументи: тренувальний і тестовий DataFrame, назву категоріального стовпця для кодування, назву цільового стовпця та параметр згладжування alpha. Вона повертає два значення: нову ознаку для тренувального і тестового DataFrame відповідно.

Інструкції 1/3

undefined XP
    1
    2
    3
  • Потрібно додати згладжування, щоб уникнути перенавчання. Тож додайте параметр \(\alpha\) до знаменника в обчисленнях train_statistics.
  • Потрібно обробити нові категорії в тестових даних. Тож передайте глобальне середнє як аргумент методу fillna().