-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathvoting.yaml
27 lines (26 loc) · 850 Bytes
/
voting.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
main:
voting: custom # values: custom, sklearn
voting:
multinomilnb:
module: sklearn.naive_bayes
class: MultinomialNB
description: Multinomial NB classifier from SciKit
bernouillinb:
module: sklearn.naive_bayes
class: BernoulliNB
description: Bernouilli NB classifier from SciKit
logisticregression:
module: sklearn.linear_model
class: LogisticRegression
description: Logistic Regression classifier from SciKit
sgd:
module: sklearn.linear_model
class: SGDClassifier
description: SGD classifier from SciKit
linearrsvc:
module: sklearn.svm
class: LinearSVC
description: Linear SVC classifier from SciKit
# ,
# "nusvc": (NuSVC(), "nusvc",
# "Nu SVC classifier from SciKit")