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

Added onnx export support for several multiclass classifiers #4462

Open
wants to merge 2 commits into
base: master
from

Conversation

@harishsk
Copy link
Member

harishsk commented Nov 8, 2019

This PR adds Onnx export support + tests for the following multiclass classifiers:

  • LbfgsMaximumEntropyMulticlassTrainer
  • LightGbmMulticlassTrainer
  • LightGbmMulticlassTrainer with SoftMax
  • OneVersusAllTrainer
  • SdcaMaximumEntropyMulticlassTrainer
  • SdcaNonCalibratedMulticlassTrainer

There is a question marked with a REVIEW comment in the PR below. Please comment if you know of a better way to do what is done there.

@harishsk harishsk requested a review from dotnet/mlnet-core as a code owner Nov 8, 2019
@harishsk harishsk force-pushed the harishsk:multiclassonnx branch from 7984d5b to 0bef75f Nov 13, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
1 participant
You can’t perform that action at this time.