Как сериализовать ML модель?

Middle
235 просмотров
AFK Offer AI

joblib — стандарт для sklearn. joblib.dump(model, "model.pkl"); model = joblib.load("model.pkl"). Сохраняет Pipeline целиком (scaler + model). pickle — встроенный, но joblib эффективнее для numpy arrays. ONNX — универсальный формат, работает в разных runtime. skl2onnx — конвертация sklearn → ONNX. MLflow — логирует модель + параметры + метрики. mlflow.sklearn.save_model(). BentoML — packaging для деплоя. Безопасность: не загружай pickle из непроверенных источников — arbitrary code execution. Версионируй: модель + версия sklearn + данные обучения.

Следующий вопрос

Чем отличаются black и isort?