Сравнение моделей машинного обучения для прогнозирования клинических рисков.
Сравнительный конвейер машинного обучения с SMOTE, перекрестной проверкой и оценкой нескольких моделей.
Этот Python-фреймворк с открытым исходным кодом сравнивает алгоритмы контролируемого машинного обучения для прогнозирования риска сердечного приступа на основе демографических, образных и клинических данных пациентов. Он подходит дата-саентистам и исследователям в области здравоохранения, создающим предиктивные диагностические модели.
Он принимает записи пациентов, обрабатывает дисбаланс классов с помощью SMOTE и весов классов, выполняет оптимизацию гиперпараметров и запускает пятиблочную перекрестную проверку. Вы оцениваете модели по таким метрикам, как точность, полнота, F1-мера и ROC-AUC, вместо того чтобы полагаться только на точность.
Он подходит дата-саентистам и медицинским аналитикам, разрабатывающим модели прогнозирования клинических рисков на Python. Он не подходит нетехническим клиницистам или командам, которым требуется готовое веб-приложение для продакшна.
Оценивает логистическую регрессию, случайный лес, дерево решений, градиентный бустинг, LightGBM, AdaBoost и многослойный перцептрон (MLP).
Применяет SMOTE и взвешивание классов для улучшения обнаружения редкого класса для событий высокого сердечного риска.
Настраивает конфигурации моделей наряду с пятиблочной перекрестной проверкой для снижения переобучения.
Измеряет производительность с помощью точности, аккуратности, полноты, F1-меры, ROC-AUC и матриц ошибок.
Код доступен для скачивания, хотя в репозитории не указана конкретная лицензия open source. В README не упомянуты платные хостинговые варианты, поэтому единственные затраты — это ваше время на его запуск.
876 открытых удалённых вакансий по обучению ИИ, обновлено 15 ноября 2026 г.; в большинстве оплата $20–$170 в час. Из них 2 — для носителей русского языка. Zekai публикует их бесплатно; отклик уходит на платформу работодателя.
Он использует набор данных для прогнозирования риска сердечного приступа, полученный с Kaggle и содержащий демографические, физиологические и клинические переменные.
Фреймворк сравнивает логистическую регрессию, случайный лес, дерево решений, градиентный бустинг, LightGBM, AdaBoost и нейронные сети MLP.
Он использует метод синтетического передискретизации меньшинства (SMOTE) и веса классов в процессе обучения.
Zekai первым в Google