__init__.py 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. """
  2. The :mod:`sklearn.ensemble` module includes ensemble-based methods for
  3. classification, regression and anomaly detection.
  4. """
  5. from ._bagging import BaggingClassifier, BaggingRegressor
  6. from ._base import BaseEnsemble
  7. from ._forest import (
  8. ExtraTreesClassifier,
  9. ExtraTreesRegressor,
  10. RandomForestClassifier,
  11. RandomForestRegressor,
  12. RandomTreesEmbedding,
  13. )
  14. from ._gb import GradientBoostingClassifier, GradientBoostingRegressor
  15. from ._hist_gradient_boosting.gradient_boosting import (
  16. HistGradientBoostingClassifier,
  17. HistGradientBoostingRegressor,
  18. )
  19. from ._iforest import IsolationForest
  20. from ._stacking import StackingClassifier, StackingRegressor
  21. from ._voting import VotingClassifier, VotingRegressor
  22. from ._weight_boosting import AdaBoostClassifier, AdaBoostRegressor
  23. __all__ = [
  24. "BaseEnsemble",
  25. "RandomForestClassifier",
  26. "RandomForestRegressor",
  27. "RandomTreesEmbedding",
  28. "ExtraTreesClassifier",
  29. "ExtraTreesRegressor",
  30. "BaggingClassifier",
  31. "BaggingRegressor",
  32. "IsolationForest",
  33. "GradientBoostingClassifier",
  34. "GradientBoostingRegressor",
  35. "AdaBoostClassifier",
  36. "AdaBoostRegressor",
  37. "VotingClassifier",
  38. "VotingRegressor",
  39. "StackingClassifier",
  40. "StackingRegressor",
  41. "HistGradientBoostingClassifier",
  42. "HistGradientBoostingRegressor",
  43. ]