如何创建一个可以运行多个(八个)二进制分类模型并输出(八个)布尔预测数组的ONNX模型?

时间:2020-08-17 20:46:14

标签: python scikit-learn classification onnx

使用scikit Learn(pyspark / azure数据块环境)使用XGBoost拥有一个单一的二进制分类模型,该模型被转换为ONNX模型(skl2onnx),然后由我们的工程师部署到设备中并进行预测(布尔值1或0)

但是,我现在需要有8个这样的模型(相似但略有不同的输入值),因此我将获得8个不同的输出布尔值。我希望将其包含在一个ONNX模型中,以供工程师部署并获取八个预测值的数组。

是否可以让一个ONNX模型接受所有输入,然后输出来自多个分类模型的一系列预测结果?

0 个答案:

没有答案