Skip to content

Instantly share code, notes, and snippets.

@securetorobert

securetorobert/poly_features_lasso_boston.py

Created July 12, 2018 01:11
Show Gist options
  • Save securetorobert/ef48bcacac6176d014bb8a475e77efa1 to your computer and use it in GitHub Desktop.
Save securetorobert/ef48bcacac6176d014bb8a475e77efa1 to your computer and use it in GitHub Desktop.
Download ZIP
Polynomial features with lasso regression on Boston housing data
Raw
poly_features_lasso_boston.py
steps = [
('scalar', StandardScaler()),
('poly', PolynomialFeatures(degree=2)),
('model', Lasso(alpha=0.3, fit_intercept=True))
]
lasso_pipe = Pipeline(steps)
lasso_pipe.fit(X_train, y_train)
print('Training score: {}'.format(lasso_pipe.score(X_train, y_train)))
print('Test score: {}'.format(lasso_pipe.score(X_test, y_test)))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment