Installation

pip install mlw

Python 3.10+. The core package includes 11 Rust-native algorithms with no external dependencies.

Optional extras

pip install "mlw[xgboost]"
pip install "mlw[lightgbm]"
pip install "mlw[catboost]"
pip install "mlw[plots]"
pip install "mlw[all]"

Extras add XGBoost, LightGBM, CatBoost, matplotlib plots, or all of the above.

install.packages("remotes")
remotes::install_github("epagogy/ml", subdir = "r")

R ≥ 4.1.0. Requires a Rust toolchain for the native backend.

Optional backends

Install these R packages for additional algorithms:

install.packages(c("xgboost", "ranger", "glmnet", "kknn", "e1071", "naivebayes"))

Verify

import ml
print(ml.__version__)
library(ml)
packageVersion("ml")
Quickstart →