Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- space = hp.choice('classifier_type', [
- {
- 'type': 'naive_bayes',
- },
- {
- 'type': 'svm',
- 'C': hp.lognormal('svm_C', 0, 1),
- 'kernel': hp.choice('svm_kernel', [
- {'ktype': 'linear'},
- {'ktype': 'RBF', 'width': hp.lognormal('svm_rbf_width', 0, 1)},
- ]),
- },
- {
- 'type': 'dtree',
- 'criterion': hp.choice('dtree_criterion', ['gini', 'entropy']),
- 'max_depth': hp.choice('dtree_max_depth',
- [None, hp.qlognormal('dtree_max_depth_int', 3, 1, 1)]),
- 'min_samples_split': hp.qlognormal('dtree_min_samples_split', 2, 1, 1),
- },
- ])
Add Comment
Please, Sign In to add comment