Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # can add in other models here, just add to search_space below
- log_reg = {"classifier": [LogisticRegression()],
- "classifier__penalty": ['l2','l1'],
- "classifier__C": np.logspace(0, 4, 10)}
- svm = {"classifier": [SVC()],
- "classifier__C": [0.01,0.1,1,10,100],
- "classifier__kernel":['linear','poly','rbf','sigmoid']}
- log_reg_solver = {"classifier": [LogisticRegression()],
- "classifier__penalty": ['l2'],
- "classifier__C": np.logspace(0, 4, 10),
- "classifier__solver":['newton-cg','saga','sag','liblinear']}
- rand_forest = {"classifier": [RandomForestClassifier()],
- "classifier__n_estimators": [10, 100, 1000],
- "classifier__max_depth":[5,8,15,25,30,None],
- "classifier__min_samples_leaf":[1,2,5,10,15,100],
- "classifier__max_leaf_nodes": [2, 5,10]}
- search_space = [svm, log_reg, log_reg_solver, rand_forest]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement