-
使用ONNX进行模型推理演示的项目:onnx-ml-demo
资源介绍
ONNX-ML演示
火车
使用综合数据训练模型:
python3 train/train.py
模型文件将在models目录中生成。
推理
这将从models目录中加载模型,并对综合数据进行推断:
python3 train/infer.py
使用ONNXRuntime转换模型并运行推理
scikit学习模型
转换模型
python3 converter/convert_basic.py models/scaler.pkl models/scaler.onnx
python3 converter/convert_basic.py models/clf.pkl models/clf.onnx
在没有ZipMap运算符的情况下转换RandomForestClassifier模型
python3 converter/convert_basic.py models/clf.pkl models