Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ADD detectree #1184

Open
kyakuno opened this issue Jul 24, 2023 · 3 comments
Open

ADD detectree #1184

kyakuno opened this issue Jul 24, 2023 · 3 comments

Comments

@kyakuno
Copy link
Collaborator

kyakuno commented Jul 24, 2023

https://github.com/martibosch/detectree

@Koichi73
Copy link
Contributor

Koichi73 commented Dec 2, 2024

お世話になります。

本モデルはLightGBMをベースとしておりますが、
ailia-modelsではLightGBM(scikit-learn)をベースとしたモデルの追加を想定しているかお伺いしたく存じます。

質問背景:

  1. 以下のリポジトリで、scikit-learnの例がない
    https://github.com/axinc-ai/export-to-onnx
  2. 本モデルをONNXにエクスポートした際に含まれるオペレータ(TreeEnsembleClassifier)が、ailiaでサポートされていない
    (エラーの詳細を次のコメントに記載します)

LightGBMをベースとしたモデルの追加を想定している場合は、TreeEnsembleClassifierを使わない形でのエクスポート方法について検討したいと考えております。
お手数をおかけしますが、ご確認のほどよろしくお願いいたします。

@Koichi73
Copy link
Contributor

Koichi73 commented Dec 2, 2024

エラー詳細

ailia.core.AiliaUnsupportNetException: code: -9 (Unsupported network.)
+ error detail : Layer:LGBMClassifier(TreeEnsembleClassifier) Error:Unsupported layer: TreeEnsembleClassifier

※参考
ONNXエクスポート方法:ドキュメント
TreeEnsembleClassifier:https://onnx.ai/onnx/operators/onnx_aionnxml_TreeEnsembleClassifier.html

@Koichi73
Copy link
Contributor

Koichi73 commented Dec 5, 2024

こちら、ドキュメントの「ailia SDKでの推論でエラーが発生する場合」を見落としていました。申し訳ございません。

onnxruntimeを使用して、サンプルを作成いたします。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants