__init__.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. """
  2. The :mod:`sklearn.feature_selection` module implements feature selection
  3. algorithms. It currently includes univariate filter selection methods and the
  4. recursive feature elimination algorithm.
  5. """
  6. from ._base import SelectorMixin
  7. from ._from_model import SelectFromModel
  8. from ._mutual_info import mutual_info_classif, mutual_info_regression
  9. from ._rfe import RFE, RFECV
  10. from ._sequential import SequentialFeatureSelector
  11. from ._univariate_selection import (
  12. GenericUnivariateSelect,
  13. SelectFdr,
  14. SelectFpr,
  15. SelectFwe,
  16. SelectKBest,
  17. SelectPercentile,
  18. chi2,
  19. f_classif,
  20. f_oneway,
  21. f_regression,
  22. r_regression,
  23. )
  24. from ._variance_threshold import VarianceThreshold
  25. __all__ = [
  26. "GenericUnivariateSelect",
  27. "SequentialFeatureSelector",
  28. "RFE",
  29. "RFECV",
  30. "SelectFdr",
  31. "SelectFpr",
  32. "SelectFwe",
  33. "SelectKBest",
  34. "SelectFromModel",
  35. "SelectPercentile",
  36. "VarianceThreshold",
  37. "chi2",
  38. "f_classif",
  39. "f_oneway",
  40. "f_regression",
  41. "r_regression",
  42. "mutual_info_classif",
  43. "mutual_info_regression",
  44. "SelectorMixin",
  45. ]