search

NIRMA-PATIENT-INTAKE / disease

MIT License
3 stars 1 forks source link

Применение связки "AutoML и объясняющая модель" #8

Open nicl-nno opened 1 year ago

nicl-nno commented 1 year ago

Добрый день,

Посмотрел на вашу реализацию и подумал, что возможно тут можно применить часть возможность создаваемого нами AutoML-решения FEDOT (https://github.com/aimclub/FEDOT).

FEDOT может построить ансамблевую модель для решения задачи классификаци+ построить доп. объясняющую модель на основе дерева решений. При этом можно задать какую-то кастомную целевую функцию - заточенную на понижение доли false-negative, например.

Возможно, вам будет интересно его попробовать в дополнение к вашей базовой модели лог. регрессии - чтобы повысить точность, не проиграв в интерпретируемости.

EgorovM commented 1 year ago

Привет! Очень интересное предложение! Мы сейчас как раз на стадии тестирования алгоритмов, мы попробуем добавить в пайплайн fedot, спасибо! Потом обязательно сообщим, что получилось

nicl-nno commented 11 months ago

Добрый день,

Увидел статью https://habr.com/ru/articles/752580/ - а какая в итоге модель получилась после применения FEDOT? Удивил перекос в recall.

nicl-nno commented 11 months ago

Посмотрел по коду - если для вас целевой критерий это f1, то стоит и AutoML запускать указывая его. Сейчас видимо по умолчанию используется roc_auc.