Турнир среди моделей аплифта: тестирование S-learner, T-learner, X-learner и Causal Forest на синтетических и полусинтетических данных с метрикой PEHE
Аналитик из Avito провел турнир, сравнивая четыре популярные модели аплифта (S-learner, T-learner, X-learner и Causal Forest) на синтетических и полусинтетических данных с использованием метрики PEHE. В статье объясняются архитектуры моделей, их сильные и слабые стороны через футбольную аналогию, кульминацией которой стали групповой этап и олимпийская система плей-офф. Код доступен на GitHub для воспроизведения.
Статья, написанная Григорием Крюковым, аналитиком отдела доверия и безопасности Avito, представляет чемпионат среди четырех моделей аплифтинга: S-learner, T-learner, X-learner и Causal Forest. Используя аналогию с чемпионатом мира по футболу, автор описывает архитектуру каждой модели: S-learner обучает единую модель с обработкой как признаком; T-learner обучает отдельные модели для групп
Показать ещё ↓
Источник: Habr — хаб ML —
оригинал