-
Notifications
You must be signed in to change notification settings - Fork 52
/
__init__.py
40 lines (33 loc) · 1.32 KB
/
__init__.py
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
28
29
30
31
32
33
34
35
36
37
38
39
40
"""
This module contains a collection of models that implement a simple function:
:func:`~revscoring.ScorerModel.score`. Currently, all models are
a subclass of :class:`revscoring.scorer_models.MLScorerModel`
which means that they also implement
:meth:`~revscoring.scorer_models.MLScorerModel.train` and
:meth:`~revscoring.scorer_models.MLScorerModel.test` methods.
Support Vector Classifiers
++++++++++++++++++++++++++
.. automodule:: revscoring.scorer_models.svc
Naive Bayes Classifiers
+++++++++++++++++++++++
.. automodule:: revscoring.scorer_models.nb
Random Forest
+++++++++++++
.. automodule:: revscoring.scorer_models.rf
Abstract classes
++++++++++++++++
.. automodule:: revscoring.scorer_models.scorer_model
"""
from .svc import SVC, SVCModel, LinearSVC, LinearSVCModel, RBFSVC, RBFSVCModel
from .nb import (NB, NBModel, GaussianNB, GaussianNBModel, MultinomialNB,
MultinomialNBModel, BernoulliNB, BernoulliNBModel)
from .scorer_model import ScorerModel, MLScorerModel
from .sklearn_classifier import ScikitLearnClassifier
from .rf import RF, RFModel
__all__ = [
SVC, SVCModel, LinearSVC, LinearSVCModel, RBFSVC, RBFSVCModel,
NB, NBModel, GaussianNB, GaussianNBModel, MultinomialNB,
MultinomialNBModel, BernoulliNB, BernoulliNBModel,
ScorerModel, MLScorerModel, ScikitLearnClassifier,
RF, RFModel
]