Skip to content

Instantly share code, notes, and snippets.

@risenW
Last active July 10, 2019 11:43
Show Gist options
  • Save risenW/72cadaefc926d4c3aa2ac326a4d4e4cd to your computer and use it in GitHub Desktop.
Save risenW/72cadaefc926d4c3aa2ac326a4d4e4cd to your computer and use it in GitHub Desktop.
#Import single models
from sklearn.linear_model import LinearRegression, LogisticRegression
from sklearn.svm import SVR, SVC
from sklearn.neighbors import KNeighborsRegressor,KNeighborsClassifier
#Classification models
log_cf = LogisticRegression(solver='lbfgs', random_state=rand_seed)
svc_cf = SVC(gamma='scale', random_state=rand_seed)
knn_cf = KNeighborsClassifier()
classification_models = [log_cf, svc_cf, knn_cf]
#Regression models
linear_reg = LinearRegression()
svr_reg = SVR(gamma='scale')
knn_reg = KNeighborsRegressor()
regression_models = [linear_reg, svr_reg, knn_reg]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment