Machine Learning

scikit-learn does supervised and unsupervised learning, plus the tools around it: preprocessing, model selection, and evaluation. Every estimator learns with fit. Put your preprocessing and model in one Pipeline, and run cross-validation and searches on that, since the pipeline keeps test data out of training. You may not need a separate boosting library: its histogram-based gradient boosting is inspired by LightGBM and supports missing values and categorical data natively. Deep learning is another matter: it doesn't fit scikit-learn's design constraints, so it's out of scope.

pgmpy does causal and probabilistic reasoning with graphical models. Its quickstart follows the tasks: learn a graph's structure from data, fit the distributions for a graph you know, then query the model for probabilities or causal effects.

Feature-engine is designed to work with dataframes: a DataFrame goes in, and the same DataFrame comes out, with no column order or name changes. Instead of wrapping transformers in a ColumnTransformer, you tell each one which variables to transform. Its transformers work just like any scikit-learn transformer, so they fit in a Pipeline.

XGBoost is an optimized distributed gradient boosting library. Its Python package has a native interface, a scikit-learn interface, and a Dask interface.

LightGBM is designed for faster training, lower memory use, and large-scale data. Pass category columns as they are: LightGBM uses them directly, with no one-hot encoding.

CatBoost aims for great results with default parameters, and it takes non-numeric features without you turning them into numbers. Name the category columns in cat_features, and don't one-hot encode them, which hurts both training speed and quality.

Prophet fits an additive model of the trend, yearly, weekly, and daily seasonality, and holiday effects. It works best with series that have strong seasonal effects and several seasons of history. It follows the scikit-learn model API: pass fit a DataFrame with ds and y columns, then call predict.

StatsForecast offers widely used univariate models, like automatic ARIMA, ETS, and Theta, optimized for speed, plus benchmark models for baselines. Data goes in as a long-format DataFrame with unique_id, ds, and y columns, and for many series the docs recommend the forecast method.

sktime gives forecasting, classification, clustering, and anomaly detection one unified interface, with interfaces to scikit-learn and Prophet. Split the series with temporal_train_test_split, set a ForecastingHorizon, then fit and predict, as in Get Started. Through reduction, a scikit-learn regressor can solve a forecasting task.

TimesFM is a pretrained foundation model from Google Research for time-series forecasting. It forecasts zero-shot, with no training on your data: load a checkpoint and call predict with a horizon. Check the license of the weights you download: some are for non-commercial, non-production use only.

To evaluate a model on time-series data, test it on observations from after its training data. scikit-learn's TimeSeriesSplit does that in cross-validation. For more, see awesome-machine-learning.