The most popular Python library for Machine Learning with extensive documentation and community support.
Machine Learning in Python
13 projects in AI & ML. Most starred: scikit-learn, timesfm and xgboost.
scikit-learn is the starting point for classical machine learning on tabular data. It gives every model the same fit and predict interface, plus preprocessing, pipelines, cross-validation and metrics. Many libraries follow its API, so they plug into the same pipelines: imbalanced-learn adds resampling for skewed classes, Feature-engine adds feature transformers, and sktime applies the same style to time series.
General4
Resampling techniques for imbalanced datasets, compatible with scikit-learn.
A Python library for causal and probabilistic reasoning with graphical models.
Sklearn compatible API with the widest toolset for feature engineering and selection.
Gradient Boosting3
xgboostC++ core
A scalable, portable, and distributed gradient boosting library.
lightgbmC++ core
A fast, distributed, high performance gradient boosting framework.
catboostC++ core
A fast, scalable, high performance gradient boosting on decision trees library.
Time Series Forecasting4
A tool for producing forecasts for time series with multiple seasonality and trend changes.
Fast statistical forecasting models such as ARIMA, ETS, and Theta, compiled with numba.
A unified scikit-learn-style framework for forecasting and other time-series learning tasks.
A pretrained foundation model from Google Research for time-series forecasting, with non-commercial default weights.
Hyperparameter Optimization2
A hyperparameter optimization framework with define-by-run search spaces and pruning of unpromising trials.
Distributed asynchronous hyperparameter optimization with Tree-structured Parzen Estimators.