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.

On structured data, gradient boosting usually beats other model families. XGBoost, LightGBM and CatBoost all train boosted decision trees but differ in details: LightGBM is known for fast training on large datasets, CatBoost handles categorical features natively, and XGBoost is the most widely deployed. All three accept scikit-learn style estimators, so trying two of them on your data is cheap. Optuna and Hyperopt tune their hyperparameters, with Optuna's define-by-run API and trial pruning making it the common pick for new projects.

For forecasting, Prophet fits business series with trend, seasonality and holiday effects, StatsForecast runs classical models such as ARIMA and ETS quickly across many series, and TimesFM is a pretrained foundation model from Google Research that forecasts without per-series training. pgmpy covers Bayesian networks and causal inference. Start forecasting with a simple statistical baseline before moving to heavier models, because seasonal naive and ETS forecasts are often hard to beat and make it clear whether a more complex model is earning its cost.

General4

scikit-learn

The most popular Python library for Machine Learning with extensive documentation and community support.

67.9kupdated today1.9.1

imbalanced-learn

Resampling techniques for imbalanced datasets, compatible with scikit-learn.

7.1kupdated 3mo ago0.14.2

pgmpy

A Python library for causal and probabilistic reasoning with graphical models.

3.4kupdated 7d ago1.1.2

feature-engine

Sklearn compatible API with the widest toolset for feature engineering and selection.

2.3kupdated 1mo ago1.9.4

Gradient Boosting3

xgboostC++ core

A scalable, portable, and distributed gradient boosting library.

28.9kupdated 1d ago3.4.2

lightgbmC++ core

A fast, distributed, high performance gradient boosting framework.

18.9kupdated today4.7.0

catboostC++ core

A fast, scalable, high performance gradient boosting on decision trees library.

9.2kupdated today1.2.10

Time Series Forecasting4

prophet

A tool for producing forecasts for time series with multiple seasonality and trend changes.

20.5kupdated 2d ago1.5.0

statsforecast

Fast statistical forecasting models such as ARIMA, ETS, and Theta, compiled with numba.

4.9kupdated 2d ago2.1.1

sktime

A unified scikit-learn-style framework for forecasting and other time-series learning tasks.

10.1kupdated today1.2.0

timesfm

A pretrained foundation model from Google Research for time-series forecasting, with non-commercial default weights.

34.8kupdated 25d ago3.0.0

Hyperparameter Optimization2

optuna

A hyperparameter optimization framework with define-by-run search spaces and pruning of unpromising trials.

15kupdated 1d ago5.0.0

hyperopt

Distributed asynchronous hyperparameter optimization with Tree-structured Parzen Estimators.

7.6kupdated 5d ago0.3.0
Esc

Type to search every entry. ↑↓ to move, Enter to open.