onnx/sklearn-onnx

해결하는 문제

scikit-learn 모델을 Open Neural Network Exchange (ONNX) 형식으로 변환할 수 있는 방법을 제공합니다. 이를 통해 사용자는 ONNX Runtime과 같은 도구를 사용하여 훈련 환경에서 고성능 스코어링 환경으로 모델을 이동시켜 추론 속도와 이식성을 향상시킬 수 있습니다.

작동 방식

이 라이브러리는 scikit-learn 모델과 변환기를 ONNX 연산자로 변환합니다. 다양한 지원되는 scikit-learn 모델을 지원하며, 다른 라이브러리의 모델이나 변환기를 scikit-learn 파이프라인 내에서 처리하기 위해 외부 컨버터 등록 기능도 제공합니다.

대상 사용자

scikit-learn으로 모델을 훈련하지만 고성능 추론 엔진이 필요한 프로덕션 환경에 배포해야 하는 데이터 과학자 및 ML 엔지니어.

주요 특징

• scikit-learn 모델을 ONNX 형식으로 변환 가능.
• 고성능 스코어링을 위한 ONNX Runtime과 호환.
• 사용자 정의 파이프라인을 위한 외부 컨버터 등록 지원.
• 최신 ONNX opset 21 지원.