Add prediction_scaling to rescale predictions toward the training target distribution - #1261

Closed

psinger-prior wants to merge 4 commits into

mainfrom

prediction-scaling

Commits on Sep 10, 2026

Add prediction_scaling: rescale predictions toward the training target distribution
psinger-prior
committed
f3b201a

Commits on Sep 11, 2026

Add changelog fragments for prediction_scaling
psinger-prior
committed
b8242ee
Use the raw target for the regressor's sampler weights on the differentiable path
psinger-prior
committed
b9bc4b2
Test prediction scaling as a fixed per-row function instead of across batch sizes
psinger-prior
committed
16b3495