Fix a quadratic algorithm in MachineBranchProbabilityInfo.