Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
Fixes onnx exports for binary classification trainers #4463
This PR fixes the issues with exporting the following binary classification trainers to onnx:
Note: I couldn't create consistent regression models, even when seeding, so I didn't include baseline comparison on these tests.
Update: Baseline files have changed. Waiting for @harishsk's PRs to go through before submitting my changes.