search

microsoft / hummingbird

Hummingbird compiles trained ML models into tensor computation for faster inference.
MIT License
3.32k stars 274 forks source link

Fix SGDClassifier predict result bug #703

Closed ytwei3 closed 1 year ago

ytwei3 commented 1 year ago

The bug is described in #702.

Here is one possible solution. I made changes to the code logic, so that the original prediction result should be computed in advance instead of being calculated at the end using the output (which is actually the result of predict_prob)

ytwei3 commented 1 year ago

@microsoft-github-policy-service agree

interesaaat commented 1 year ago

Thanks @hikari234 for the contribution!