Atelier Digital
Автоматизация

ML TCAD — автоматизация машинного обучения

Собственная разработка
AI / Data Science 4 месяца Опубликован

О проекте

Веб-платформа для автоматизации процессов машинного обучения (AutoML). Позволяет пользователям загружать данные, а система автоматически подбирает лучшие модели для задач классификации, регрессии и прогнозирования временных рядов. Включает предобработку данных, отбор признаков, оптимизацию гиперпараметров, валидацию и деплой.

Запрос клиента

Полная автоматизация ML

Создать платформу для автоматизации обучения моделей ML для задач классификации, регрессии и временных рядов

Сложности

  • Реализовать универсальный AutoML пайплайн
  • Обеспечить гибкость для разных типов данных
  • Оптимизировать время обучения
  • Сделать систему доступной для нетехнических специалистов

Цели

  • Снизить порог входа в ML
  • Ускорить эксперименты в 10 раз
  • Обеспечить высокое качество моделей
  • Автоматизировать полный цикл ML-разработки

Наше решение

AutoML платформа TCAD

Комплексная система автоматизации всех этапов ML: от данных до деплоя

Подход

  • Реализован модуль автоматической предобработки данных (очистка, масштабирование, кодирование)
  • Внедрена система автоматического отбора признаков
  • Создан бэкенд для обучения моделей (XGBoost, LightGBM, Neural Networks)
  • Реализована оптимизация гиперпараметров (Bayesian, Grid Search)
  • Добавлена автоматическая валидация и выбор лучшей модели
  • Создан API для интеграции обученных моделей

Технологии

Python FastAPI PostgreSQL Scikit-learn XGBoost LightGBM PyTorch Optuna React Docker Kubernetes MLflow

Результаты

  • Поддержка 30+ алгоритмов
  • Возможность загрузки данных в 10+ форматах
  • Интеграция с MLflow и Kubeflow
  • Сокращение времени обучения с дней до часов