__init__.py 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. """
  2. The :mod:`sklearn.linear_model` module implements a variety of linear models.
  3. """
  4. # See http://scikit-learn.sourceforge.net/modules/sgd.html and
  5. # http://scikit-learn.sourceforge.net/modules/linear_model.html for
  6. # complete documentation.
  7. from ._base import LinearRegression
  8. from ._bayes import ARDRegression, BayesianRidge
  9. from ._coordinate_descent import (
  10. ElasticNet,
  11. ElasticNetCV,
  12. Lasso,
  13. LassoCV,
  14. MultiTaskElasticNet,
  15. MultiTaskElasticNetCV,
  16. MultiTaskLasso,
  17. MultiTaskLassoCV,
  18. enet_path,
  19. lasso_path,
  20. )
  21. from ._glm import GammaRegressor, PoissonRegressor, TweedieRegressor
  22. from ._huber import HuberRegressor
  23. from ._least_angle import (
  24. Lars,
  25. LarsCV,
  26. LassoLars,
  27. LassoLarsCV,
  28. LassoLarsIC,
  29. lars_path,
  30. lars_path_gram,
  31. )
  32. from ._logistic import LogisticRegression, LogisticRegressionCV
  33. from ._omp import (
  34. OrthogonalMatchingPursuit,
  35. OrthogonalMatchingPursuitCV,
  36. orthogonal_mp,
  37. orthogonal_mp_gram,
  38. )
  39. from ._passive_aggressive import PassiveAggressiveClassifier, PassiveAggressiveRegressor
  40. from ._perceptron import Perceptron
  41. from ._quantile import QuantileRegressor
  42. from ._ransac import RANSACRegressor
  43. from ._ridge import Ridge, RidgeClassifier, RidgeClassifierCV, RidgeCV, ridge_regression
  44. from ._sgd_fast import Hinge, Huber, Log, ModifiedHuber, SquaredLoss
  45. from ._stochastic_gradient import SGDClassifier, SGDOneClassSVM, SGDRegressor
  46. from ._theil_sen import TheilSenRegressor
  47. __all__ = [
  48. "ARDRegression",
  49. "BayesianRidge",
  50. "ElasticNet",
  51. "ElasticNetCV",
  52. "Hinge",
  53. "Huber",
  54. "HuberRegressor",
  55. "Lars",
  56. "LarsCV",
  57. "Lasso",
  58. "LassoCV",
  59. "LassoLars",
  60. "LassoLarsCV",
  61. "LassoLarsIC",
  62. "LinearRegression",
  63. "Log",
  64. "LogisticRegression",
  65. "LogisticRegressionCV",
  66. "ModifiedHuber",
  67. "MultiTaskElasticNet",
  68. "MultiTaskElasticNetCV",
  69. "MultiTaskLasso",
  70. "MultiTaskLassoCV",
  71. "OrthogonalMatchingPursuit",
  72. "OrthogonalMatchingPursuitCV",
  73. "PassiveAggressiveClassifier",
  74. "PassiveAggressiveRegressor",
  75. "Perceptron",
  76. "QuantileRegressor",
  77. "Ridge",
  78. "RidgeCV",
  79. "RidgeClassifier",
  80. "RidgeClassifierCV",
  81. "SGDClassifier",
  82. "SGDRegressor",
  83. "SGDOneClassSVM",
  84. "SquaredLoss",
  85. "TheilSenRegressor",
  86. "enet_path",
  87. "lars_path",
  88. "lars_path_gram",
  89. "lasso_path",
  90. "orthogonal_mp",
  91. "orthogonal_mp_gram",
  92. "ridge_regression",
  93. "RANSACRegressor",
  94. "PoissonRegressor",
  95. "GammaRegressor",
  96. "TweedieRegressor",
  97. ]