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 をサポート。