Installation¶

Requirements¶

Python 3.11 or later

Standard install¶

uv pip install h2ml
# or
pip install h2ml

With boosting libraries¶

LightGBM, XGBoost, and CatBoost are optional. Install the [boosting] extra to include them:

uv pip install h2ml[boosting]
# or
pip install h2ml[boosting]

Without this extra, the pipeline runs all sklearn models only. The registry detects which libraries are available at import time — no configuration required.

With spatial inference¶

The geo module (h2ml.geo.geo_predict) depends on h2mare, available on PyPI via the [geo] extra:

uv pip install h2ml[geo]
# or
pip install h2ml[geo]

Development setup¶

git clone https://github.com/h2ugoparra/h2ml
cd h2ml
uv sync --group dev
uv run pytest        # run the test suite
uv run mkdocs serve  # preview the docs locally

Running tests across Python versions¶

uv python install 3.11 3.12 3.13
uv run tox