希洛!
我正在尝试将分段的 logreg 模型打包到 onnx,这个想法是根据字符串值计算相应的逻辑回归,比如 if
segment = 'segm_1'
logreg_1.predict()
如何在 ONNX 图中创建选择器。我只找到 If 节点,但它只适用于用于分支的预制标志。如果有很多段,它看起来像不好的解决方案
我尝试使用 onnx python API 构建图形,但我发现缺少运算符