isn't r2 the default for regression in sklearn?