2011-11-12 18:29:38 +08:00
|
|
|
"""
|
2011-11-30 17:30:13 +08:00
|
|
|
The :mod:`sklearn.ensemble` module includes ensemble-based methods for
|
2011-11-12 19:19:16 +08:00
|
|
|
classification and regression.
|
2011-11-12 18:29:38 +08:00
|
|
|
"""
|
|
|
|
|
|
2011-11-17 02:35:57 +08:00
|
|
|
from .base import BaseEnsemble
|
2011-11-12 18:29:38 +08:00
|
|
|
from .forest import RandomForestClassifier
|
|
|
|
|
from .forest import RandomForestRegressor
|
2012-11-16 03:20:45 +08:00
|
|
|
from .forest import RandomTreesEmbedding
|
2011-11-12 18:29:38 +08:00
|
|
|
from .forest import ExtraTreesClassifier
|
|
|
|
|
from .forest import ExtraTreesRegressor
|
2013-07-23 18:59:32 +08:00
|
|
|
from .bagging import BaggingClassifier
|
|
|
|
|
from .bagging import BaggingRegressor
|
2013-01-11 17:12:53 +08:00
|
|
|
from .weight_boosting import AdaBoostClassifier
|
2012-12-06 22:52:24 +08:00
|
|
|
from .weight_boosting import AdaBoostRegressor
|
2011-09-23 22:31:08 +08:00
|
|
|
from .gradient_boosting import GradientBoostingClassifier
|
|
|
|
|
from .gradient_boosting import GradientBoostingRegressor
|
2015-04-21 07:27:30 +08:00
|
|
|
from .voting_classifier import VotingClassifier
|
2012-11-11 01:51:34 +08:00
|
|
|
|
2013-07-23 18:59:32 +08:00
|
|
|
from . import bagging
|
2012-11-11 01:51:34 +08:00
|
|
|
from . import forest
|
2011-12-30 11:17:30 +08:00
|
|
|
from . import weight_boosting
|
2012-11-11 01:51:34 +08:00
|
|
|
from . import gradient_boosting
|
2012-11-21 05:35:38 +08:00
|
|
|
from . import partial_dependence
|
2012-12-22 20:02:50 +08:00
|
|
|
|
2013-07-23 18:59:32 +08:00
|
|
|
__all__ = ["BaseEnsemble",
|
|
|
|
|
"RandomForestClassifier", "RandomForestRegressor",
|
2012-12-22 20:02:50 +08:00
|
|
|
"RandomTreesEmbedding", "ExtraTreesClassifier",
|
2013-07-23 18:59:32 +08:00
|
|
|
"ExtraTreesRegressor", "BaggingClassifier",
|
|
|
|
|
"BaggingRegressor", "GradientBoostingClassifier",
|
2013-01-11 20:33:51 +08:00
|
|
|
"GradientBoostingRegressor", "AdaBoostClassifier",
|
2015-04-21 07:27:30 +08:00
|
|
|
"AdaBoostRegressor", "VotingClassifier",
|
2015-01-26 13:33:44 +08:00
|
|
|
"bagging", "forest", "gradient_boosting",
|
2013-02-09 05:10:40 +08:00
|
|
|
"partial_dependence", "weight_boosting"]
|