search

openvinotoolkit / openvino

OpenVINO™ is an open-source toolkit for optimizing and deploying AI inference
https://docs.openvino.ai
Apache License 2.0
6.31k stars 2.08k forks source link

[Feature Request]: Add support for ZipMap and Tree Ensemble Classifier #23778

Open AbhijithGanesh opened 3 months ago

AbhijithGanesh commented 3 months ago

Request Description

I was trying to implement a CatBoost model inferencing via ONNX and ran into this error:

RuntimeError: Exception from src/inference/src/cpp/core.cpp:92:
Check 'error_message.empty()' failed at src/frontends/onnx/frontend/src/frontend.cpp:122:
FrontEnd API failed with GeneralFailure:
OpenVINO does not support the following ONNX operations: ai.onnx.ml.ZipMap, ai.onnx.ml.TreeEnsembleClassifier

Feature Use Case

This would extend support for Catboost and Tree Ensemble Classifier

Issue submission checklist

AbhijithGanesh commented 3 months ago

image

krish1209 commented 3 months ago

Hi. You can develop a custom logic for the unsupported operations, or try checking the ONNX opset version. If it is supported in a lower Opset version.

AbhijithGanesh commented 3 months ago

I am not sure if it is supported by a lower OpSet version, thats why I wanted to raise this issue and add support

AbhijithGanesh commented 3 months ago

@gkrivor Can I work alongside you on this?

AyuK03 commented 3 months ago

Request Description

I was trying to implement a CatBoost model inferencing via ONNX and ran into this error:

RuntimeError: Exception from src/inference/src/cpp/core.cpp:92:
Check 'error_message.empty()' failed at src/frontends/onnx/frontend/src/frontend.cpp:122:
FrontEnd API failed with GeneralFailure:
OpenVINO does not support the following ONNX operations: ai.onnx.ml.ZipMap, ai.onnx.ml.TreeEnsembleClassifier

Feature Use Case

This would extend support for Catboost and Tree Ensemble Classifier

Issue submission checklist

  • [x] The feature request or improvement must be related to OpenVINO

Request Description

I was trying to implement a CatBoost model inferencing via ONNX and ran into this error:

RuntimeError: Exception from src/inference/src/cpp/core.cpp:92:
Check 'error_message.empty()' failed at src/frontends/onnx/frontend/src/frontend.cpp:122:
FrontEnd API failed with GeneralFailure:
OpenVINO does not support the following ONNX operations: ai.onnx.ml.ZipMap, ai.onnx.ml.TreeEnsembleClassifier

Feature Use Case

This would extend support for Catboost and Tree Ensemble Classifier

Issue submission checklist

  • [x] The feature request or improvement must be related to OpenVINO

I would like to take up this issue.

AyuK03 commented 3 months ago

.take

github-actions[bot] commented 3 months ago

Thanks for being interested in this issue. It looks like this ticket is already assigned to a contributor. Please communicate with the assigned contributor to confirm the status of the issue.

AyuK03 commented 3 months ago

@AbhijithGanesh is this issue still open? I would like to work on it

AbhijithGanesh commented 3 months ago

Yeah I am exploring how to add these features. I'm yet to open my draft PR, I'll do that soon

mlukasze commented 3 months ago

@gkrivor could you advice, please?