Skip to content

Instantly share code, notes, and snippets.

@terrah27
Created February 16, 2019 18:08
Show Gist options
  • Save terrah27/d038a80e3b2b8160942f2d3a78c35795 to your computer and use it in GitHub Desktop.
Save terrah27/d038a80e3b2b8160942f2d3a78c35795 to your computer and use it in GitHub Desktop.
# prepare for modeling
X_train = train.drop(['id', 'target'], axis=1)
y_train = train['target']
X_test = test.drop(['id'], axis=1)
# scaling data
scaler = StandardScaler()
X_train = scaler.fit_transform(X_train)
X_test = scaler.transform(X_test)
# define models
ridge = linear_model.Ridge()
lasso = linear_model.Lasso()
elastic = linear_model.ElasticNet()
lasso_lars = linear_model.LassoLars()
bayesian_ridge = linear_model.BayesianRidge()
logistic = linear_model.LogisticRegression(solver='liblinear')
sgd = linear_model.SGDClassifier()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment