mindsdb / lightwood

Lightwood is Legos for Machine Learning.
GNU General Public License v3.0
434 stars 92 forks source link

[ENH] Weighted regression #1210

Closed QuantumPlumber closed 3 months ago

QuantumPlumber commented 3 months ago

This PR covers numeric dataset weighting for XGBoost and LightGBM, and associated tests. Users can now provide a json dictionary of histogram bin edges and associated weights, which will be used to map the labels to weights. The histogram bins are left edge -- overflows are assigned to the last bin's weight, and underflows to the first bin's weight.