You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
When using skl2onnx for exporting pipelines using HistGradientBoostingRegressor, we are seeing significant discrepancies in the output (double-digit relative error on some inputs) even when we use the CastTransformer to float32-cast inputs to the regressor. For reference, our pipeline looks something like
My suspicion is that the way histogram-based gradient boosting generates buckets and decision thresholds requires the same float64->float32 threshold adjustment that the normal GradientBoostingRegressor gets, even when the inputs are float32.