ENH: small optimization to the regression criterion