onnxruntime
c37d2728 - Implement TreeEnsemble for opset(ai.onnx.ml)==3 (#10821)

Commit
3 years ago
Implement TreeEnsemble for opset(ai.onnx.ml)==3 (#10821) * Implement TreeEnsemble for opset(ai.onnx.ml)==3 * use of InlineVector * refactoring * improve attributes retrieval * avoid creating a temporary buffer * modifies onnx.ml.cpu.json * use unordered_map * update docs/OperatorKernels.md * address PR comments (TH -> ThresholdType, ORT_RETURN...) * add a python unit test to load a TreeEnsembleRegressor following ai.onnx.ml==3 specifications
Author
Parents
Loading