ONNX
(Open Neural Network Exchange
) 可以优化机器学习模型的推理,
许多框架(包括 TensorFlow
、PyTorch
、SciKit-Learn
、Keras
、Chainer
、MXNet
、MATLAB
和 SparkML
)中的模型都可以导出或转换为标准 ONNX
格式。
下面将以sklearn
模块中的LogisticRegression
模型为例,做一个简单介绍:
示例1
Step 1: 训练模型
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression
# 加载数据
iris = load_iris()
X, y = iris.data, iris.target
X_train, X_test, y_train, y_test = train_test_split(