Add target clipping and update init point for platt transform

amzn / pecos

PECOS - Prediction for Enormous and Correlated Spaces

https://libpecos.org/

Apache License 2.0

516 stars 105 forks source link

Closed jiong-zhang closed 11 months ago

jiong-zhang commented 11 months ago

Issue #, if available:

Description of changes:

Add target probability clipping in fit_platt_transform
Update initial point for platt transform fit. These modifications will prevent most cases where the Newton method fails.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.