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

Predictions of converted LightGbm ONNX model completely mismatched ML.Net equivalent #5549

Closed
Tracked by #6360 ...
mentorfloat opened this issue Dec 11, 2020 · 1 comment
Assignees

Comments

@mentorfloat
Copy link

System information

  • OS version: 10.0.19041
  • .NET SDKs: 3.1.402, 5.0.100-preview.7.20366.6

Issue

  • What did you do?
    Convert three trained ML.Net models (LightGbm, SdcaMaximumEntropy, LbfgsMaximumEntropy) to ONNX models with ConvertToOnnx and make predictions with 100 randomly selected data, using both the original ML.Net models and the converted ONNX models.

  • What happened?
    When making predictions with the converted LightGbm ONNX model, none of the predicted values match the predicted values made using the original ML.Net model. To test further, I repeated the pipeline with different trainers, SdcaMaximumEntropy and LbfgsMaximumEntropy. The converted ONNX models for these two returned predicted values that match the predictions made using their ML.Net equivalents.

  • What did you expect?
    LightGbm's converted ONNX model generate the same predictions as its ML.Net counterpart. Similar to how the other two trainers' did.

  • Side notes:
    Initially I couldn't get LightGbm to convert to ONNX, but that's resolved now. Closed Issue ConvertToOnnx with LightGbm Model Throws KeyNotFoundException #5542 , but likely unrelated to the above.

@luisquintanilla
Copy link
Contributor

ML.NET 3.0 has been upgraded to a new version of LightGBM which may have fixed this issue. Please open a new issue if it persists.

@github-actions github-actions bot locked and limited conversation to collaborators Jan 7, 2024
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants