Skip to content

Instantly share code, notes, and snippets.

@mohcinemadkour
Forked from terrah27/baseline_models.py
Created August 12, 2019 14:47
Show Gist options
  • Save mohcinemadkour/077a46b8ce1b786a34aab1bad33b56fb to your computer and use it in GitHub Desktop.
Save mohcinemadkour/077a46b8ce1b786a34aab1bad33b56fb 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