onnx/sklearn-onnx

解決的問題

提供將 scikit-learn 模型轉換為 Open Neural Network Exchange (ONNX) 格式的途徑。這讓使用者能利用 ONNX Runtime 等工具，將模型從訓練環境移轉至高效率評分環境，提升推論速度與可移植性。

作用方式

此套件將 scikit-learn 模型與轉換器轉換為 ONNX 操作符。支援廣泛的 scikit-learn 模型，並允許註冊外部轉換器，以在 scikit-learn 流水線中處理來自其他套件的模型或轉換器。

適用對象

使用 scikit-learn 訓練模型，但需在需要高效率推論引擎的生產環境中部署模型的資料科學家與機器學習工程師。

主要特色

• 將 scikit-learn 模型轉換為 ONNX 格式。
• 與 ONNX Runtime 相容，實現高效率評分。
• 支援外部轉換器註冊，適用於自訂流水線。
• 支援最新的 ONNX opset 21。