登录 注册
当前位置:主页 > 资源下载 > 50 > 使用ONNX进行模型推理演示的项目:onnx-ml-demo

使用ONNX进行模型推理演示的项目:onnx-ml-demo

  • 更新:2024-10-18 18:56:56
  • 大小:69KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

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