文件名称：onnx-ml-demo:使用ONNX进行模型推理的演示
文件大小：69KB
文件格式：ZIP
更新时间：2024-04-11 10:04:48
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)