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.

Sign up for GitHub

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

Jump to bottom

[BUG] cuml binary classification models do not observe threshold #351

Open
RAMitchell opened this issue Mar 28, 2023 · 1 comment
Open

[BUG] cuml binary classification models do not observe threshold #351

RAMitchell opened this issue Mar 28, 2023 · 1 comment

Comments

@RAMitchell
Copy link
Contributor

RAMitchell commented Mar 28, 2023 •
edited
Loading

When setting threshold=0.9, cuml models for CPU or GPU do not appear to observe the threshold.

Other model types seem to be working.
@RAMitchell RAMitchell changed the title [BUG] cuml classification models do not observe threshold [BUG] cuml binary classification models do not observe threshold Mar 28, 2023
@RAMitchell
Copy link
Contributor Author

This is occurring because these binary classification models are stored by treelite as vector leaf models, which are handled differently in postprocessing code and do not account for thresholds.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@RAMitchell