Nice small example, the regret column makes the point clearly. A practical extension is to report decision regret or top-1 action agreement next to MSE on a held-out set, and also check the rank of the true best action in the predicted costs. Would weighting the MSE toward the low-cost actions the planner actually considers close the gap, or does it just move the failure elsewhere?