diff --git a/sklearn_bench/elasticnet.py b/sklearn_bench/elasticnet.py index 89e820e6f..3467e0dda 100755 --- a/sklearn_bench/elasticnet.py +++ b/sklearn_bench/elasticnet.py @@ -28,7 +28,7 @@ def main(): # Create our regression object regr = ElasticNet(fit_intercept=params.fit_intercept, l1_ratio=params.l1_ratio, alpha=params.alpha, tol=params.tol, - max_iter=params.maxiter, copy_X=False) + max_iter=params.maxiter) # Time fit fit_time, _ = bench.measure_function_time(regr.fit, X_train, y_train, params=params) @@ -63,7 +63,7 @@ def main(): if __name__ == "__main__": parser = argparse.ArgumentParser(description='scikit-learn elastic-net regression ' 'benchmark') - parser.add_argument('--no-fit-intercept', dest='fit_intercept', default=False, + parser.add_argument('--no-fit-intercept', dest='fit_intercept', default=True, action='store_false', help="Don't fit intercept (assume data already centered)") parser.add_argument('--alpha', dest='alpha', type=float, default=1.0, diff --git a/sklearn_bench/lasso.py b/sklearn_bench/lasso.py index 6346d5e8a..c167bc359 100755 --- a/sklearn_bench/lasso.py +++ b/sklearn_bench/lasso.py @@ -27,7 +27,7 @@ def main(): # Create our regression object regr = Lasso(fit_intercept=params.fit_intercept, alpha=params.alpha, - tol=params.tol, max_iter=params.maxiter, copy_X=False) + tol=params.tol, max_iter=params.maxiter) # Time fit fit_time, _ = bench.measure_function_time(regr.fit, X_train, y_train, params=params) @@ -63,7 +63,7 @@ def main(): if __name__ == "__main__": parser = argparse.ArgumentParser(description='scikit-learn lasso regression ' 'benchmark') - parser.add_argument('--no-fit-intercept', dest='fit_intercept', default=False, + parser.add_argument('--no-fit-intercept', dest='fit_intercept', default=True, action='store_false', help="Don't fit intercept (assume data already centered)") parser.add_argument('--alpha', dest='alpha', type=float, default=1.0, diff --git a/sklearn_bench/linear.py b/sklearn_bench/linear.py index b97d49a6e..7da0dba45 100644 --- a/sklearn_bench/linear.py +++ b/sklearn_bench/linear.py @@ -27,7 +27,7 @@ def main(): # Create our regression object regr = LinearRegression(fit_intercept=params.fit_intercept, - n_jobs=params.n_jobs, copy_X=False) + n_jobs=params.n_jobs) # Time fit fit_time, _ = bench.measure_function_time(regr.fit, X_train, y_train, params=params)