【文件属性】:
文件名称:onnx-ml-demo:使用ONNX进行模型推理的演示
文件大小:69KB
文件格式:ZIP
更新时间:2021-03-22 16:18:08
Python
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
【文件预览】:
onnx-ml-demo-main
----train()
--------map_pred.py(295B)
--------train.py(2KB)
--------infer.py(901B)
----assets()
--------flamegraph.svg(302KB)
----cpp()
--------infer_basic.cc(11KB)
--------Makefile(263B)
--------viterbi.cc(7KB)
--------stopwatch.h(1010B)
----LICENSE(7KB)
----requirements.txt(98B)
----.gitignore(67B)
----converter()
--------viterbi.py(2KB)
--------inference_basic.py(590B)
--------convert_hmm_naive.py(2KB)
--------skl2onnx_pipeline.py(8KB)
--------visualize.py(539B)
--------skl2onnx_hmm.py(3KB)
--------utils.py(1KB)
--------convert_pipeline.py(1KB)
--------inference_pipeline.py(765B)
--------inference_hmm.py(725B)
--------convert_basic.py(794B)
--------convert_hmm_skl2onnx.py(970B)
----README.md(2KB)