46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""
|
|
The :mod:`sklearn.ensemble` module includes ensemble-based methods for
|
|
classification, regression and anomaly detection.
|
|
"""
|
|
|
|
from ._bagging import BaggingClassifier, BaggingRegressor
|
|
from ._base import BaseEnsemble
|
|
from ._forest import (
|
|
ExtraTreesClassifier,
|
|
ExtraTreesRegressor,
|
|
RandomForestClassifier,
|
|
RandomForestRegressor,
|
|
RandomTreesEmbedding,
|
|
)
|
|
from ._gb import GradientBoostingClassifier, GradientBoostingRegressor
|
|
from ._hist_gradient_boosting.gradient_boosting import (
|
|
HistGradientBoostingClassifier,
|
|
HistGradientBoostingRegressor,
|
|
)
|
|
from ._iforest import IsolationForest
|
|
from ._stacking import StackingClassifier, StackingRegressor
|
|
from ._voting import VotingClassifier, VotingRegressor
|
|
from ._weight_boosting import AdaBoostClassifier, AdaBoostRegressor
|
|
|
|
__all__ = [
|
|
"BaseEnsemble",
|
|
"RandomForestClassifier",
|
|
"RandomForestRegressor",
|
|
"RandomTreesEmbedding",
|
|
"ExtraTreesClassifier",
|
|
"ExtraTreesRegressor",
|
|
"BaggingClassifier",
|
|
"BaggingRegressor",
|
|
"IsolationForest",
|
|
"GradientBoostingClassifier",
|
|
"GradientBoostingRegressor",
|
|
"AdaBoostClassifier",
|
|
"AdaBoostRegressor",
|
|
"VotingClassifier",
|
|
"VotingRegressor",
|
|
"StackingClassifier",
|
|
"StackingRegressor",
|
|
"HistGradientBoostingClassifier",
|
|
"HistGradientBoostingRegressor",
|
|
]
|