AutoGluon Tabular - Quick Start

In this tutorial, we will see how to use AutoGluon’s TabularPredictor to predict the values of a target column based on the other columns in a tabular dataset.

Begin by making sure AutoGluon is installed, and then import AutoGluon’s TabularDataset and TabularPredictor. We will use the former to load data and the latter to train models and make predictions.

!python -m pip install --upgrade pip
!python -m pip install autogluon

Show code cell output

Hide code cell output

Requirement already satisfied: pip in /home/ci/opt/venv/lib/python3.12/site-packages (25.1.1)
Requirement already satisfied: autogluon in /home/ci/opt/venv/lib/python3.12/site-packages (1.3.2b20250716)
Requirement already satisfied: autogluon.core==1.3.2b20250716 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core[all]==1.3.2b20250716->autogluon) (1.3.2b20250716)
Requirement already satisfied: autogluon.features==1.3.2b20250716 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon) (1.3.2b20250716)
Requirement already satisfied: autogluon.tabular==1.3.2b20250716 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.3.2b20250716)
Requirement already satisfied: autogluon.multimodal==1.3.2b20250716 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon) (1.3.2b20250716)
Requirement already satisfied: autogluon.timeseries==1.3.2b20250716 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.timeseries[all]==1.3.2b20250716->autogluon) (1.3.2b20250716)
Requirement already satisfied: numpy<2.4.0,>=1.25.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (2.1.3)
Requirement already satisfied: scipy<1.17,>=1.5.4 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (1.16.0)
Requirement already satisfied: scikit-learn<1.8.0,>=1.4.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (1.6.1)
Requirement already satisfied: networkx<4,>=3.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (3.5)
Requirement already satisfied: pandas<2.4.0,>=2.0.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (2.3.1)
Requirement already satisfied: tqdm<5,>=4.38 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (4.67.1)
Requirement already satisfied: requests in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (2.32.4)
Requirement already satisfied: matplotlib<3.11,>=3.7.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (3.10.3)
Requirement already satisfied: boto3<2,>=1.10 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (1.39.7)
Requirement already satisfied: autogluon.common==1.3.2b20250716 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (1.3.2b20250716)
Requirement already satisfied: pyarrow<21.0.0,>=7.0.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.common==1.3.2b20250716->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (20.0.0)
Requirement already satisfied: psutil<7.1.0,>=5.7.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.common==1.3.2b20250716->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (7.0.0)
Requirement already satisfied: joblib<1.7,>=1.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.common==1.3.2b20250716->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (1.5.1)
Requirement already satisfied: ray<2.45,>=2.10.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (2.44.1)
Requirement already satisfied: hyperopt<0.2.8,>=0.2.7 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.core[all]==1.3.2b20250716->autogluon) (0.2.7)
Requirement already satisfied: Pillow<12,>=10.0.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (11.3.0)
Requirement already satisfied: torch<2.8,>=2.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (2.7.1)
Requirement already satisfied: lightning<2.8,>=2.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (2.5.2)
Requirement already satisfied: transformers<4.50,>=4.38.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from transformers[sentencepiece]<4.50,>=4.38.0->autogluon.multimodal==1.3.2b20250716->autogluon) (4.49.0)
Requirement already satisfied: accelerate<2.0,>=0.34.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (1.9.0)
Requirement already satisfied: jsonschema<4.24,>=4.18 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (4.23.0)
Requirement already satisfied: seqeval<1.3.0,>=1.2.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (1.2.2)
Requirement already satisfied: evaluate<0.5.0,>=0.4.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (0.4.5)
Requirement already satisfied: timm<1.0.7,>=0.9.5 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (1.0.3)
Requirement already satisfied: torchvision<0.23.0,>=0.16.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (0.22.1)
Requirement already satisfied: scikit-image<0.26.0,>=0.19.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (0.25.2)
Requirement already satisfied: text-unidecode<1.4,>=1.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (1.3)
Requirement already satisfied: torchmetrics<1.8,>=1.2.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (1.7.4)
Requirement already satisfied: omegaconf<2.4.0,>=2.1.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (2.3.0)
Requirement already satisfied: pytorch-metric-learning<2.9,>=1.3.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (2.8.1)
Requirement already satisfied: nlpaug<1.2.0,>=1.1.10 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (1.1.11)
Requirement already satisfied: nltk<3.10,>=3.4.5 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (3.9.1)
Requirement already satisfied: openmim<0.4.0,>=0.3.7 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (0.3.9)
Requirement already satisfied: defusedxml<0.7.2,>=0.7.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (0.7.1)
Requirement already satisfied: jinja2<3.2,>=3.0.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (3.1.6)
Requirement already satisfied: tensorboard<3,>=2.9 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (2.19.0)
Requirement already satisfied: pytesseract<0.4,>=0.3.9 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (0.3.13)
Requirement already satisfied: nvidia-ml-py3<8.0,>=7.352.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (7.352.0)
Requirement already satisfied: pdf2image<1.19,>=1.17.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.multimodal==1.3.2b20250716->autogluon) (1.17.0)
Requirement already satisfied: fastai<2.9,>=2.3.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.tabular[all]==1.3.2b20250716->autogluon) (2.8.2)
Requirement already satisfied: lightgbm<4.7,>=4.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.tabular[all]==1.3.2b20250716->autogluon) (4.6.0)
Requirement already satisfied: spacy<3.9 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.tabular[all]==1.3.2b20250716->autogluon) (3.8.7)
Requirement already satisfied: xgboost<3.1,>=2.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.tabular[all]==1.3.2b20250716->autogluon) (3.0.2)
Requirement already satisfied: huggingface_hub[torch] in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.tabular[all]==1.3.2b20250716->autogluon) (0.33.4)
Requirement already satisfied: pytabkit<1.6,>=1.5 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.5.2)
Requirement already satisfied: catboost<1.3,>=1.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.2.8)
Requirement already satisfied: einops<0.9,>=0.7 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.tabular[all]==1.3.2b20250716->autogluon) (0.8.1)
Requirement already satisfied: pytorch_lightning in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (2.5.2)
Requirement already satisfied: gluonts<0.17,>=0.15.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.16.2)
Requirement already satisfied: statsforecast<2.0.2,>=1.7.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (2.0.1)
Requirement already satisfied: mlforecast<0.15.0,>=0.14.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.14.0)
Requirement already satisfied: utilsforecast<0.2.12,>=0.2.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.2.11)
Requirement already satisfied: coreforecast<0.0.17,>=0.0.12 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.0.16)
Requirement already satisfied: fugue>=0.9.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.9.1)
Requirement already satisfied: orjson~=3.9 in /home/ci/opt/venv/lib/python3.12/site-packages (from autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (3.11.0)
Requirement already satisfied: packaging>=20.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from accelerate<2.0,>=0.34.0->autogluon.multimodal==1.3.2b20250716->autogluon) (25.0)
Requirement already satisfied: pyyaml in /home/ci/opt/venv/lib/python3.12/site-packages (from accelerate<2.0,>=0.34.0->autogluon.multimodal==1.3.2b20250716->autogluon) (6.0.2)
Requirement already satisfied: safetensors>=0.4.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from accelerate<2.0,>=0.34.0->autogluon.multimodal==1.3.2b20250716->autogluon) (0.5.3)
Requirement already satisfied: botocore<1.40.0,>=1.39.7 in /home/ci/opt/venv/lib/python3.12/site-packages (from boto3<2,>=1.10->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (1.39.7)
Requirement already satisfied: jmespath<2.0.0,>=0.7.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from boto3<2,>=1.10->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (1.0.1)
Requirement already satisfied: s3transfer<0.14.0,>=0.13.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from boto3<2,>=1.10->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (0.13.0)
Requirement already satisfied: python-dateutil<3.0.0,>=2.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from botocore<1.40.0,>=1.39.7->boto3<2,>=1.10->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (2.9.0.post0)
Requirement already satisfied: urllib3!=2.2.0,<3,>=1.25.4 in /home/ci/opt/venv/lib/python3.12/site-packages (from botocore<1.40.0,>=1.39.7->boto3<2,>=1.10->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (2.5.0)
Requirement already satisfied: graphviz in /home/ci/opt/venv/lib/python3.12/site-packages (from catboost<1.3,>=1.2->autogluon.tabular[all]==1.3.2b20250716->autogluon) (0.21)
Requirement already satisfied: plotly in /home/ci/opt/venv/lib/python3.12/site-packages (from catboost<1.3,>=1.2->autogluon.tabular[all]==1.3.2b20250716->autogluon) (6.2.0)
Requirement already satisfied: six in /home/ci/opt/venv/lib/python3.12/site-packages (from catboost<1.3,>=1.2->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.17.0)
Requirement already satisfied: datasets>=2.0.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.2b20250716->autogluon) (2.14.4)
Requirement already satisfied: dill in /home/ci/opt/venv/lib/python3.12/site-packages (from evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.2b20250716->autogluon) (0.3.7)
Requirement already satisfied: xxhash in /home/ci/opt/venv/lib/python3.12/site-packages (from evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.2b20250716->autogluon) (3.5.0)
Requirement already satisfied: multiprocess in /home/ci/opt/venv/lib/python3.12/site-packages (from evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.2b20250716->autogluon) (0.70.15)
Requirement already satisfied: fsspec>=2021.05.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from fsspec[http]>=2021.05.0->evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.2b20250716->autogluon) (2025.7.0)
Requirement already satisfied: pip in /home/ci/opt/venv/lib/python3.12/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.2b20250716->autogluon) (25.1.1)
Requirement already satisfied: fastdownload<2,>=0.0.5 in /home/ci/opt/venv/lib/python3.12/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.2b20250716->autogluon) (0.0.7)
Requirement already satisfied: fastcore<1.9,>=1.8.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.8.5)
Requirement already satisfied: fasttransform>=0.0.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.2b20250716->autogluon) (0.0.2)
Requirement already satisfied: fastprogress>=0.2.4 in /home/ci/opt/venv/lib/python3.12/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.0.3)
Requirement already satisfied: plum-dispatch in /home/ci/opt/venv/lib/python3.12/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.2b20250716->autogluon) (2.5.7)
Requirement already satisfied: cloudpickle in /home/ci/opt/venv/lib/python3.12/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.2b20250716->autogluon) (3.1.1)
Requirement already satisfied: pydantic<3,>=1.7 in /home/ci/opt/venv/lib/python3.12/site-packages (from gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (2.11.7)
Requirement already satisfied: toolz~=0.10 in /home/ci/opt/venv/lib/python3.12/site-packages (from gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.12.1)
Requirement already satisfied: typing-extensions~=4.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (4.14.1)
Requirement already satisfied: future in /home/ci/opt/venv/lib/python3.12/site-packages (from hyperopt<0.2.8,>=0.2.7->autogluon.core[all]==1.3.2b20250716->autogluon) (1.0.0)
Requirement already satisfied: py4j in /home/ci/opt/venv/lib/python3.12/site-packages (from hyperopt<0.2.8,>=0.2.7->autogluon.core[all]==1.3.2b20250716->autogluon) (0.10.9.9)
Requirement already satisfied: MarkupSafe>=2.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from jinja2<3.2,>=3.0.3->autogluon.multimodal==1.3.2b20250716->autogluon) (3.0.2)
Requirement already satisfied: attrs>=22.2.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from jsonschema<4.24,>=4.18->autogluon.multimodal==1.3.2b20250716->autogluon) (25.3.0)
Requirement already satisfied: jsonschema-specifications>=2023.03.6 in /home/ci/opt/venv/lib/python3.12/site-packages (from jsonschema<4.24,>=4.18->autogluon.multimodal==1.3.2b20250716->autogluon) (2025.4.1)
Requirement already satisfied: referencing>=0.28.4 in /home/ci/opt/venv/lib/python3.12/site-packages (from jsonschema<4.24,>=4.18->autogluon.multimodal==1.3.2b20250716->autogluon) (0.36.2)
Requirement already satisfied: rpds-py>=0.7.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from jsonschema<4.24,>=4.18->autogluon.multimodal==1.3.2b20250716->autogluon) (0.26.0)
Requirement already satisfied: lightning-utilities<2.0,>=0.10.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from lightning<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (0.14.3)
Requirement already satisfied: aiohttp!=4.0.0a0,!=4.0.0a1 in /home/ci/opt/venv/lib/python3.12/site-packages (from fsspec[http]>=2021.05.0->evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.2b20250716->autogluon) (3.12.14)
Requirement already satisfied: setuptools in /home/ci/opt/venv/lib/python3.12/site-packages (from lightning-utilities<2.0,>=0.10.0->lightning<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (80.9.0)
Requirement already satisfied: contourpy>=1.0.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from matplotlib<3.11,>=3.7.0->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (1.3.2)
Requirement already satisfied: cycler>=0.10 in /home/ci/opt/venv/lib/python3.12/site-packages (from matplotlib<3.11,>=3.7.0->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (0.12.1)
Requirement already satisfied: fonttools>=4.22.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from matplotlib<3.11,>=3.7.0->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (4.59.0)
Requirement already satisfied: kiwisolver>=1.3.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from matplotlib<3.11,>=3.7.0->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (1.4.8)
Requirement already satisfied: pyparsing>=2.3.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from matplotlib<3.11,>=3.7.0->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (3.2.3)
Requirement already satisfied: numba in /home/ci/opt/venv/lib/python3.12/site-packages (from mlforecast<0.15.0,>=0.14.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.61.2)
Requirement already satisfied: optuna in /home/ci/opt/venv/lib/python3.12/site-packages (from mlforecast<0.15.0,>=0.14.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (4.4.0)
Requirement already satisfied: window-ops in /home/ci/opt/venv/lib/python3.12/site-packages (from mlforecast<0.15.0,>=0.14.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.0.15)
Requirement already satisfied: gdown>=4.0.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from nlpaug<1.2.0,>=1.1.10->autogluon.multimodal==1.3.2b20250716->autogluon) (5.2.0)
Requirement already satisfied: click in /home/ci/opt/venv/lib/python3.12/site-packages (from nltk<3.10,>=3.4.5->autogluon.multimodal==1.3.2b20250716->autogluon) (8.2.1)
Requirement already satisfied: regex>=2021.8.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from nltk<3.10,>=3.4.5->autogluon.multimodal==1.3.2b20250716->autogluon) (2024.11.6)
Requirement already satisfied: antlr4-python3-runtime==4.9.* in /home/ci/opt/venv/lib/python3.12/site-packages (from omegaconf<2.4.0,>=2.1.1->autogluon.multimodal==1.3.2b20250716->autogluon) (4.9.3)
Requirement already satisfied: colorama in /home/ci/opt/venv/lib/python3.12/site-packages (from openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (0.4.4)
Requirement already satisfied: model-index in /home/ci/opt/venv/lib/python3.12/site-packages (from openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (0.1.11)
Requirement already satisfied: opendatalab in /home/ci/opt/venv/lib/python3.12/site-packages (from openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (0.0.10)
Requirement already satisfied: rich in /home/ci/opt/venv/lib/python3.12/site-packages (from openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (14.0.0)
Requirement already satisfied: tabulate in /home/ci/opt/venv/lib/python3.12/site-packages (from openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (0.9.0)
Requirement already satisfied: pytz>=2020.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from pandas<2.4.0,>=2.0.0->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (2025.2)
Requirement already satisfied: tzdata>=2022.7 in /home/ci/opt/venv/lib/python3.12/site-packages (from pandas<2.4.0,>=2.0.0->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (2025.2)
Requirement already satisfied: annotated-types>=0.6.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from pydantic<3,>=1.7->gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.7.0)
Requirement already satisfied: pydantic-core==2.33.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from pydantic<3,>=1.7->gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (2.33.2)
Requirement already satisfied: typing-inspection>=0.4.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from pydantic<3,>=1.7->gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.4.1)
Requirement already satisfied: filelock in /home/ci/opt/venv/lib/python3.12/site-packages (from ray<2.45,>=2.10.0->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (3.18.0)
Requirement already satisfied: msgpack<2.0.0,>=1.0.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from ray<2.45,>=2.10.0->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (1.1.1)
Requirement already satisfied: protobuf!=3.19.5,>=3.15.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from ray<2.45,>=2.10.0->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (6.31.1)
Requirement already satisfied: aiosignal in /home/ci/opt/venv/lib/python3.12/site-packages (from ray<2.45,>=2.10.0->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (1.4.0)
Requirement already satisfied: frozenlist in /home/ci/opt/venv/lib/python3.12/site-packages (from ray<2.45,>=2.10.0->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (1.7.0)
Requirement already satisfied: aiohttp_cors in /home/ci/opt/venv/lib/python3.12/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (0.8.1)
Requirement already satisfied: colorful in /home/ci/opt/venv/lib/python3.12/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (0.5.7)
Requirement already satisfied: py-spy>=0.4.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (0.4.0)
Requirement already satisfied: grpcio>=1.42.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (1.73.1)
Requirement already satisfied: opencensus in /home/ci/opt/venv/lib/python3.12/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (0.11.4)
Requirement already satisfied: prometheus_client>=0.7.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (0.22.1)
Requirement already satisfied: smart_open in /home/ci/opt/venv/lib/python3.12/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (7.3.0.post1)
Requirement already satisfied: virtualenv!=20.21.1,>=20.0.24 in /home/ci/opt/venv/lib/python3.12/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (20.31.2)
Requirement already satisfied: tensorboardX>=1.9 in /home/ci/opt/venv/lib/python3.12/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (2.6.4)
Requirement already satisfied: charset_normalizer<4,>=2 in /home/ci/opt/venv/lib/python3.12/site-packages (from requests->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (3.4.2)
Requirement already satisfied: idna<4,>=2.5 in /home/ci/opt/venv/lib/python3.12/site-packages (from requests->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (3.10)
Requirement already satisfied: certifi>=2017.4.17 in /home/ci/opt/venv/lib/python3.12/site-packages (from requests->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (2025.7.14)
Requirement already satisfied: imageio!=2.35.0,>=2.33 in /home/ci/opt/venv/lib/python3.12/site-packages (from scikit-image<0.26.0,>=0.19.1->autogluon.multimodal==1.3.2b20250716->autogluon) (2.37.0)
Requirement already satisfied: tifffile>=2022.8.12 in /home/ci/opt/venv/lib/python3.12/site-packages (from scikit-image<0.26.0,>=0.19.1->autogluon.multimodal==1.3.2b20250716->autogluon) (2025.6.11)
Requirement already satisfied: lazy-loader>=0.4 in /home/ci/opt/venv/lib/python3.12/site-packages (from scikit-image<0.26.0,>=0.19.1->autogluon.multimodal==1.3.2b20250716->autogluon) (0.4)
Requirement already satisfied: threadpoolctl>=3.1.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from scikit-learn<1.8.0,>=1.4.0->autogluon.core==1.3.2b20250716->autogluon.core[all]==1.3.2b20250716->autogluon) (3.6.0)
Requirement already satisfied: spacy-legacy<3.1.0,>=3.0.11 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (3.0.12)
Requirement already satisfied: spacy-loggers<2.0.0,>=1.0.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.0.5)
Requirement already satisfied: murmurhash<1.1.0,>=0.28.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.0.13)
Requirement already satisfied: cymem<2.1.0,>=2.0.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (2.0.11)
Requirement already satisfied: preshed<3.1.0,>=3.0.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (3.0.10)
Requirement already satisfied: thinc<8.4.0,>=8.3.4 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (8.3.6)
Requirement already satisfied: wasabi<1.2.0,>=0.9.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.1.3)
Requirement already satisfied: srsly<3.0.0,>=2.4.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (2.5.1)
Requirement already satisfied: catalogue<2.1.0,>=2.0.6 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (2.0.10)
Requirement already satisfied: weasel<0.5.0,>=0.1.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (0.4.1)
Requirement already satisfied: typer<1.0.0,>=0.3.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (0.16.0)
Requirement already satisfied: langcodes<4.0.0,>=3.2.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (3.5.0)
Requirement already satisfied: language-data>=1.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from langcodes<4.0.0,>=3.2.0->spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.3.0)
Requirement already satisfied: statsmodels>=0.13.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from statsforecast<2.0.2,>=1.7.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.14.5)
Requirement already satisfied: absl-py>=0.4 in /home/ci/opt/venv/lib/python3.12/site-packages (from tensorboard<3,>=2.9->autogluon.multimodal==1.3.2b20250716->autogluon) (2.3.1)
Requirement already satisfied: markdown>=2.6.8 in /home/ci/opt/venv/lib/python3.12/site-packages (from tensorboard<3,>=2.9->autogluon.multimodal==1.3.2b20250716->autogluon) (3.8.2)
Requirement already satisfied: tensorboard-data-server<0.8.0,>=0.7.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from tensorboard<3,>=2.9->autogluon.multimodal==1.3.2b20250716->autogluon) (0.7.2)
Requirement already satisfied: werkzeug>=1.0.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from tensorboard<3,>=2.9->autogluon.multimodal==1.3.2b20250716->autogluon) (3.1.3)
Requirement already satisfied: blis<1.4.0,>=1.3.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from thinc<8.4.0,>=8.3.4->spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.3.0)
Requirement already satisfied: confection<1.0.0,>=0.0.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from thinc<8.4.0,>=8.3.4->spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (0.1.5)
Requirement already satisfied: sympy>=1.13.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (1.14.0)
Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.6.77 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (12.6.77)
Requirement already satisfied: nvidia-cuda-runtime-cu12==12.6.77 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (12.6.77)
Requirement already satisfied: nvidia-cuda-cupti-cu12==12.6.80 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (12.6.80)
Requirement already satisfied: nvidia-cudnn-cu12==9.5.1.17 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (9.5.1.17)
Requirement already satisfied: nvidia-cublas-cu12==12.6.4.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (12.6.4.1)
Requirement already satisfied: nvidia-cufft-cu12==11.3.0.4 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (11.3.0.4)
Requirement already satisfied: nvidia-curand-cu12==10.3.7.77 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (10.3.7.77)
Requirement already satisfied: nvidia-cusolver-cu12==11.7.1.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (11.7.1.2)
Requirement already satisfied: nvidia-cusparse-cu12==12.5.4.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (12.5.4.2)
Requirement already satisfied: nvidia-cusparselt-cu12==0.6.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (0.6.3)
Requirement already satisfied: nvidia-nccl-cu12==2.26.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (2.26.2)
Requirement already satisfied: nvidia-nvtx-cu12==12.6.77 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (12.6.77)
Requirement already satisfied: nvidia-nvjitlink-cu12==12.6.85 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (12.6.85)
Requirement already satisfied: nvidia-cufile-cu12==1.11.1.6 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (1.11.1.6)
Requirement already satisfied: triton==3.3.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (3.3.1)
Requirement already satisfied: tokenizers<0.22,>=0.21 in /home/ci/opt/venv/lib/python3.12/site-packages (from transformers<4.50,>=4.38.0->transformers[sentencepiece]<4.50,>=4.38.0->autogluon.multimodal==1.3.2b20250716->autogluon) (0.21.2)
Requirement already satisfied: hf-xet<2.0.0,>=1.1.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from huggingface_hub[torch]->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.1.5)
Requirement already satisfied: sentencepiece!=0.1.92,>=0.1.91 in /home/ci/opt/venv/lib/python3.12/site-packages (from transformers[sentencepiece]<4.50,>=4.38.0->autogluon.multimodal==1.3.2b20250716->autogluon) (0.2.0)
Requirement already satisfied: shellingham>=1.3.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from typer<1.0.0,>=0.3.0->spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.5.4)
Requirement already satisfied: cloudpathlib<1.0.0,>=0.7.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from weasel<0.5.0,>=0.1.0->spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (0.21.1)
Requirement already satisfied: wrapt in /home/ci/opt/venv/lib/python3.12/site-packages (from smart_open->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (1.17.2)
Requirement already satisfied: aiohappyeyeballs>=2.5.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>=2021.05.0->evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.2b20250716->autogluon) (2.6.1)
Requirement already satisfied: multidict<7.0,>=4.5 in /home/ci/opt/venv/lib/python3.12/site-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>=2021.05.0->evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.2b20250716->autogluon) (6.6.3)
Requirement already satisfied: propcache>=0.2.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>=2021.05.0->evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.2b20250716->autogluon) (0.3.2)
Requirement already satisfied: yarl<2.0,>=1.17.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from aiohttp!=4.0.0a0,!=4.0.0a1->fsspec[http]>=2021.05.0->evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.2b20250716->autogluon) (1.20.1)
Requirement already satisfied: triad>=0.9.7 in /home/ci/opt/venv/lib/python3.12/site-packages (from fugue>=0.9.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.9.8)
Requirement already satisfied: adagio>=0.2.4 in /home/ci/opt/venv/lib/python3.12/site-packages (from fugue>=0.9.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.2.6)
Requirement already satisfied: beautifulsoup4 in /home/ci/opt/venv/lib/python3.12/site-packages (from gdown>=4.0.0->nlpaug<1.2.0,>=1.1.10->autogluon.multimodal==1.3.2b20250716->autogluon) (4.13.4)
Requirement already satisfied: marisa-trie>=1.1.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from language-data>=1.2->langcodes<4.0.0,>=3.2.0->spacy<3.9->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.2.1)
Requirement already satisfied: llvmlite<0.45,>=0.44.0dev0 in /home/ci/opt/venv/lib/python3.12/site-packages (from numba->mlforecast<0.15.0,>=0.14.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (0.44.0)
Requirement already satisfied: markdown-it-py>=2.2.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from rich->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (3.0.0)
Requirement already satisfied: pygments<3.0.0,>=2.13.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from rich->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (2.19.2)
Requirement already satisfied: mdurl~=0.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from markdown-it-py>=2.2.0->rich->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (0.1.2)
Requirement already satisfied: patsy>=0.5.6 in /home/ci/opt/venv/lib/python3.12/site-packages (from statsmodels>=0.13.2->statsforecast<2.0.2,>=1.7.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (1.0.1)
Requirement already satisfied: mpmath<1.4,>=1.1.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from sympy>=1.13.3->torch<2.8,>=2.2->autogluon.multimodal==1.3.2b20250716->autogluon) (1.3.0)
Requirement already satisfied: fs in /home/ci/opt/venv/lib/python3.12/site-packages (from triad>=0.9.7->fugue>=0.9.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (2.4.16)
Requirement already satisfied: distlib<1,>=0.3.7 in /home/ci/opt/venv/lib/python3.12/site-packages (from virtualenv!=20.21.1,>=20.0.24->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (0.3.9)
Requirement already satisfied: platformdirs<5,>=3.9.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from virtualenv!=20.21.1,>=20.0.24->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (4.3.8)
Requirement already satisfied: soupsieve>1.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from beautifulsoup4->gdown>=4.0.0->nlpaug<1.2.0,>=1.1.10->autogluon.multimodal==1.3.2b20250716->autogluon) (2.7)
Requirement already satisfied: appdirs~=1.4.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from fs->triad>=0.9.7->fugue>=0.9.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (1.4.4)
Requirement already satisfied: ordered-set in /home/ci/opt/venv/lib/python3.12/site-packages (from model-index->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (4.1.0)
Requirement already satisfied: opencensus-context>=0.1.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (0.1.3)
Requirement already satisfied: google-api-core<3.0.0,>=1.0.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (2.25.1)
Requirement already satisfied: googleapis-common-protos<2.0.0,>=1.56.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from google-api-core<3.0.0,>=1.0.0->opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (1.70.0)
Requirement already satisfied: proto-plus<2.0.0,>=1.22.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from google-api-core<3.0.0,>=1.0.0->opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (1.26.1)
Requirement already satisfied: google-auth<3.0.0,>=2.14.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from google-api-core<3.0.0,>=1.0.0->opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (2.40.3)
Requirement already satisfied: cachetools<6.0,>=2.0.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from google-auth<3.0.0,>=2.14.1->google-api-core<3.0.0,>=1.0.0->opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (5.5.2)
Requirement already satisfied: pyasn1-modules>=0.2.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from google-auth<3.0.0,>=2.14.1->google-api-core<3.0.0,>=1.0.0->opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (0.4.2)
Requirement already satisfied: rsa<5,>=3.1.4 in /home/ci/opt/venv/lib/python3.12/site-packages (from google-auth<3.0.0,>=2.14.1->google-api-core<3.0.0,>=1.0.0->opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (4.9.1)
Requirement already satisfied: pyasn1>=0.1.3 in /home/ci/opt/venv/lib/python3.12/site-packages (from rsa<5,>=3.1.4->google-auth<3.0.0,>=2.14.1->google-api-core<3.0.0,>=1.0.0->opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.2b20250716->autogluon) (0.6.1)
Requirement already satisfied: pycryptodome in /home/ci/opt/venv/lib/python3.12/site-packages (from opendatalab->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (3.23.0)
Requirement already satisfied: openxlab in /home/ci/opt/venv/lib/python3.12/site-packages (from opendatalab->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.2b20250716->autogluon) (0.0.11)
Requirement already satisfied: alembic>=1.5.0 in /home/ci/opt/venv/lib/python3.12/site-packages (from optuna->mlforecast<0.15.0,>=0.14.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (1.16.4)
Requirement already satisfied: colorlog in /home/ci/opt/venv/lib/python3.12/site-packages (from optuna->mlforecast<0.15.0,>=0.14.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (6.9.0)
Requirement already satisfied: sqlalchemy>=1.4.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from optuna->mlforecast<0.15.0,>=0.14.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (2.0.41)
Requirement already satisfied: Mako in /home/ci/opt/venv/lib/python3.12/site-packages (from alembic>=1.5.0->optuna->mlforecast<0.15.0,>=0.14.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (1.3.10)
Requirement already satisfied: greenlet>=1 in /home/ci/opt/venv/lib/python3.12/site-packages (from sqlalchemy>=1.4.2->optuna->mlforecast<0.15.0,>=0.14.0->autogluon.timeseries==1.3.2b20250716->autogluon.timeseries[all]==1.3.2b20250716->autogluon) (3.2.3)
Requirement already satisfied: narwhals>=1.15.1 in /home/ci/opt/venv/lib/python3.12/site-packages (from plotly->catboost<1.3,>=1.2->autogluon.tabular[all]==1.3.2b20250716->autogluon) (1.47.0)
Requirement already satisfied: beartype>=0.16.2 in /home/ci/opt/venv/lib/python3.12/site-packages (from plum-dispatch->fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.2b20250716->autogluon) (0.21.0)
Requirement already satisfied: PySocks!=1.5.7,>=1.5.6 in /home/ci/opt/venv/lib/python3.12/site-packages (from requests[socks]->gdown>=4.0.0->nlpaug<1.2.0,>=1.1.10->autogluon.multimodal==1.3.2b20250716->autogluon) (1.7.1)

from autogluon.tabular import TabularDataset, TabularPredictor

Example Data

For this tutorial we will use a dataset from the cover story of Nature issue 7887: AI-guided intuition for math theorems. The goal is to predict a knot’s signature based on its properties. We sampled 10K training and 5K test examples from the original data. The sampled dataset make this tutorial run quickly, but AutoGluon can handle the full dataset if desired.

We load this dataset directly from a URL. AutoGluon’s TabularDataset is a subclass of pandas DataFrame, so any DataFrame methods can be used on TabularDataset as well.

data_url = 'https://raw.githubusercontent.com/mli/ag-docs/main/knot_theory/'
train_data = TabularDataset(f'{data_url}train.csv')
train_data.head()

	Unnamed: 0	chern_simons	cusp_volume	hyperbolic_adjoint_torsion_degree	hyperbolic_torsion_degree	injectivity_radius	longitudinal_translation	meridinal_translation_imag	meridinal_translation_real	short_geodesic_imag_part	short_geodesic_real_part	Symmetry_0	Symmetry_D3	Symmetry_D4	Symmetry_D6	Symmetry_D8	Symmetry_Z/2 + Z/2	volume	signature
0	70746	0.090530	12.226322	0	10	0.507756	10.685555	1.144192	-0.519157	-2.760601	1.015512	0.0	0.0	0.0	0.0	0.0	1.0	11.393225	-2
1	240827	0.232453	13.800773	0	14	0.413645	10.453156	1.320249	-0.158522	-3.013258	0.827289	0.0	0.0	0.0	0.0	0.0	1.0	12.742782	0
2	155659	-0.144099	14.761030	0	14	0.436928	13.405199	1.101142	0.768894	2.233106	0.873856	0.0	0.0	0.0	0.0	0.0	0.0	15.236505	2
3	239963	-0.171668	13.738019	0	22	0.249481	27.819496	0.493827	-1.188718	-2.042771	0.498961	0.0	0.0	0.0	0.0	0.0	0.0	17.279890	-8
4	90504	0.235188	15.896359	0	10	0.389329	15.330971	1.036879	0.722828	-3.056138	0.778658	0.0	0.0	0.0	0.0	0.0	0.0	16.749298	4

Our targets are stored in the “signature” column, which has 18 unique integers. Even though pandas didn’t correctly recognize this data type as categorical, AutoGluon will fix this issue.

label = 'signature'
train_data[label].describe()

count    10000.000000
mean        -0.022000
std          3.025166
min        -12.000000
25%         -2.000000
50%          0.000000
75%          2.000000
max         12.000000
Name: signature, dtype: float64

Training

We now construct a TabularPredictor by specifying the label column name and then train on the dataset with TabularPredictor.fit(). We don’t need to specify any other parameters. AutoGluon will recognize this is a multi-class classification task, perform automatic feature engineering, train multiple models, and then ensemble the models to create the final predictor.

predictor = TabularPredictor(label=label).fit(train_data)

Show code cell output

Hide code cell output

No path specified. Models will be saved in: "AutogluonModels/ag-20250717_003910"
Verbosity: 2 (Standard Logging)
=================== System Info ===================
AutoGluon Version:  1.3.2b20250716
Python Version:     3.12.10
Operating System:   Linux
Platform Machine:   x86_64
Platform Version:   #1 SMP Wed Mar 12 14:53:59 UTC 2025
CPU Count:          8
Memory Avail:       28.58 GB / 30.95 GB (92.3%)
Disk Space Avail:   199.89 GB / 255.99 GB (78.1%)
===================================================
No presets specified! To achieve strong results with AutoGluon, it is recommended to use the available presets. Defaulting to `'medium'`...
	Recommended Presets (For more details refer to https://auto.gluon.ai/stable/tutorials/tabular/tabular-essentials.html#presets):
	presets='experimental' : New in v1.2: Pre-trained foundation model + parallel fits. The absolute best accuracy without consideration for inference speed. Does not support GPU.
	presets='best'         : Maximize accuracy. Recommended for most users. Use in competitions and benchmarks.
	presets='high'         : Strong accuracy with fast inference speed.
	presets='good'         : Good accuracy with very fast inference speed.
	presets='medium'       : Fast training time, ideal for initial prototyping.
/home/ci/autogluon/common/src/autogluon/common/utils/utils.py:97: UserWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html. The pkg_resources package is slated for removal as early as 2025-11-30. Refrain from using this package or pin to Setuptools<81.
  import pkg_resources
Beginning AutoGluon training ...
AutoGluon will save models to "/home/ci/autogluon/docs/tutorials/tabular/AutogluonModels/ag-20250717_003910"
Train Data Rows:    10000
Train Data Columns: 18
Label Column:       signature
AutoGluon infers your prediction problem is: 'multiclass' (because dtype of label-column == int, but few unique label-values observed).
	First 10 (of 13) unique label values:  [np.int64(-2), np.int64(0), np.int64(2), np.int64(-8), np.int64(4), np.int64(-4), np.int64(-6), np.int64(8), np.int64(6), np.int64(10)]
	If 'multiclass' is not the correct problem_type, please manually specify the problem_type parameter during Predictor init (You may specify problem_type as one of: ['binary', 'multiclass', 'regression', 'quantile'])
Problem Type:       multiclass
Preprocessing data ...
Warning: Some classes in the training set have fewer than 10 examples. AutoGluon will only keep 9 out of 13 classes for training and will not try to predict the rare classes. To keep more classes, increase the number of datapoints from these rare classes in the training data or reduce label_count_threshold.
Fraction of data from classes with at least 10 examples that will be kept for training models: 0.9984
Train Data Class Count: 9
Using Feature Generators to preprocess the data ...
Fitting AutoMLPipelineFeatureGenerator...
	Available Memory:                    29263.13 MB
	Train Data (Original)  Memory Usage: 1.37 MB (0.0% of available memory)
	Inferring data type of each feature based on column values. Set feature_metadata_in to manually specify special dtypes of the features.
	Stage 1 Generators:
		Fitting AsTypeFeatureGenerator...
			Note: Converting 5 features to boolean dtype as they only contain 2 unique values.
	Stage 2 Generators:
		Fitting FillNaFeatureGenerator...
	Stage 3 Generators:
		Fitting IdentityFeatureGenerator...
	Stage 4 Generators:
		Fitting DropUniqueFeatureGenerator...
	Stage 5 Generators:
		Fitting DropDuplicatesFeatureGenerator...
	Useless Original Features (Count: 1): ['Symmetry_D8']
		These features carry no predictive signal and should be manually investigated.
		This is typically a feature which has the same value for all rows.
		These features do not need to be present at inference time.
	Types of features in original data (raw dtype, special dtypes):
		('float', []) : 14 | ['chern_simons', 'cusp_volume', 'injectivity_radius', 'longitudinal_translation', 'meridinal_translation_imag', ...]
		('int', [])   :  3 | ['Unnamed: 0', 'hyperbolic_adjoint_torsion_degree', 'hyperbolic_torsion_degree']
	Types of features in processed data (raw dtype, special dtypes):
		('float', [])     : 9 | ['chern_simons', 'cusp_volume', 'injectivity_radius', 'longitudinal_translation', 'meridinal_translation_imag', ...]
		('int', [])       : 3 | ['Unnamed: 0', 'hyperbolic_adjoint_torsion_degree', 'hyperbolic_torsion_degree']
		('int', ['bool']) : 5 | ['Symmetry_0', 'Symmetry_D3', 'Symmetry_D4', 'Symmetry_D6', 'Symmetry_Z/2 + Z/2']
	0.1s = Fit runtime
	17 features in original data used to generate 17 features in processed data.
	Train Data (Processed) Memory Usage: 0.96 MB (0.0% of available memory)
Data preprocessing and feature engineering runtime = 0.1s ...
AutoGluon will gauge predictive performance using evaluation metric: 'accuracy'
	To change this, specify the eval_metric parameter of Predictor()
Automatically generating train/validation split with holdout_frac=0.1, Train Rows: 8985, Val Rows: 999
User-specified model hyperparameters to be fit:
{
	'NN_TORCH': [{}],
	'GBM': [{'extra_trees': True, 'ag_args': {'name_suffix': 'XT'}}, {}, {'learning_rate': 0.03, 'num_leaves': 128, 'feature_fraction': 0.9, 'min_data_in_leaf': 3, 'ag_args': {'name_suffix': 'Large', 'priority': 0, 'hyperparameter_tune_kwargs': None}}],
	'CAT': [{}],
	'XGB': [{}],
	'FASTAI': [{}],
	'RF': [{'criterion': 'gini', 'ag_args': {'name_suffix': 'Gini', 'problem_types': ['binary', 'multiclass']}}, {'criterion': 'entropy', 'ag_args': {'name_suffix': 'Entr', 'problem_types': ['binary', 'multiclass']}}, {'criterion': 'squared_error', 'ag_args': {'name_suffix': 'MSE', 'problem_types': ['regression', 'quantile']}}],
	'XT': [{'criterion': 'gini', 'ag_args': {'name_suffix': 'Gini', 'problem_types': ['binary', 'multiclass']}}, {'criterion': 'entropy', 'ag_args': {'name_suffix': 'Entr', 'problem_types': ['binary', 'multiclass']}}, {'criterion': 'squared_error', 'ag_args': {'name_suffix': 'MSE', 'problem_types': ['regression', 'quantile']}}],
	'KNN': [{'weights': 'uniform', 'ag_args': {'name_suffix': 'Unif'}}, {'weights': 'distance', 'ag_args': {'name_suffix': 'Dist'}}],
}
Fitting 13 L1 models, fit_strategy="sequential" ...
Fitting model: KNeighborsUnif ...
	0.2232	 = Validation score   (accuracy)
	0.05s	 = Training   runtime
	0.01s	 = Validation runtime
Fitting model: KNeighborsDist ...
	0.2132	 = Validation score   (accuracy)
	0.02s	 = Training   runtime
	0.01s	 = Validation runtime
Fitting model: NeuralNetFastAI ...
	0.9399	 = Validation score   (accuracy)
	8.78s	 = Training   runtime
	0.01s	 = Validation runtime
Fitting model: LightGBMXT ...
	0.9459	 = Validation score   (accuracy)
	4.12s	 = Training   runtime
	0.05s	 = Validation runtime
Fitting model: LightGBM ...
	0.956	 = Validation score   (accuracy)
	3.72s	 = Training   runtime
	0.03s	 = Validation runtime
Fitting model: RandomForestGini ...
	0.9449	 = Validation score   (accuracy)
	1.5s	 = Training   runtime
	0.08s	 = Validation runtime
Fitting model: RandomForestEntr ...
	0.9499	 = Validation score   (accuracy)
	1.9s	 = Training   runtime
	0.08s	 = Validation runtime
Fitting model: CatBoost ...
	0.956	 = Validation score   (accuracy)
	19.14s	 = Training   runtime
	0.0s	 = Validation runtime
Fitting model: ExtraTreesGini ...
	0.9469	 = Validation score   (accuracy)
	1.14s	 = Training   runtime
	0.09s	 = Validation runtime
Fitting model: ExtraTreesEntr ...
	0.9429	 = Validation score   (accuracy)
	1.11s	 = Training   runtime
	0.07s	 = Validation runtime
Fitting model: XGBoost ...
	0.957	 = Validation score   (accuracy)
	5.21s	 = Training   runtime
	0.07s	 = Validation runtime
Fitting model: NeuralNetTorch ...
	0.952	 = Validation score   (accuracy)
	265.21s	 = Training   runtime
	0.01s	 = Validation runtime
Fitting model: LightGBMLarge ...
	0.9499	 = Validation score   (accuracy)
	8.84s	 = Training   runtime
	0.07s	 = Validation runtime
Fitting model: WeightedEnsemble_L2 ...
	Ensemble Weights: {'NeuralNetFastAI': 0.25, 'ExtraTreesGini': 0.25, 'RandomForestEntr': 0.2, 'XGBoost': 0.15, 'KNeighborsUnif': 0.1, 'NeuralNetTorch': 0.05}
	0.965	 = Validation score   (accuracy)
	0.12s	 = Training   runtime
	0.0s	 = Validation runtime
AutoGluon training complete, total runtime = 322.8s ... Best model: WeightedEnsemble_L2 | Estimated inference throughput: 3732.7 rows/s (999 batch size)
TabularPredictor saved. To load, use: predictor = TabularPredictor.load("/home/ci/autogluon/docs/tutorials/tabular/AutogluonModels/ag-20250717_003910")

Model fitting should take a few minutes or less depending on your CPU. You can make training faster by specifying the time_limit argument. For example, fit(..., time_limit=60) will stop training after 60 seconds. Higher time limits will generally result in better prediction performance, and excessively low time limits will prevent AutoGluon from training and ensembling a reasonable set of models.

Prediction

Once we have a predictor that is fit on the training dataset, we can load a separate set of data to use for prediction and evaulation.

test_data = TabularDataset(f'{data_url}test.csv')

y_pred = predictor.predict(test_data.drop(columns=[label]))
y_pred.head()

Loaded data from: https://raw.githubusercontent.com/mli/ag-docs/main/knot_theory/test.csv | Columns = 19 / 19 | Rows = 5000 -> 5000

0   -4
1   -2
2    0
3    4
4    2
Name: signature, dtype: int64

Evaluation

We can evaluate the predictor on the test dataset using the evaluate() function, which measures how well our predictor performs on data that was not used for fitting the models.

predictor.evaluate(test_data, silent=True)

{'accuracy': 0.9486,
 'balanced_accuracy': np.float64(0.7485562162063563),
 'mcc': np.float64(0.9369988193374609)}

AutoGluon’s TabularPredictor also provides the leaderboard() function, which allows us to evaluate the performance of each individual trained model on the test data.

predictor.leaderboard(test_data)

	model	score_test	score_val	eval_metric	pred_time_test	pred_time_val	fit_time	pred_time_test_marginal	pred_time_val_marginal	fit_time_marginal	stack_level	can_infer	fit_order
0	WeightedEnsemble_L2	0.9486	0.964965	accuracy	0.867560	0.267633	282.426446	0.007750	0.000906	0.123837	2	True	14
1	LightGBM	0.9456	0.955956	accuracy	0.155413	0.028151	3.716166	0.155413	0.028151	3.716166	1	True	5
2	XGBoost	0.9448	0.956957	accuracy	0.407917	0.069953	5.207891	0.407917	0.069953	5.207891	1	True	11
3	LightGBMLarge	0.9444	0.949950	accuracy	0.413835	0.068902	8.837025	0.413835	0.068902	8.837025	1	True	13
4	CatBoost	0.9432	0.955956	accuracy	0.026426	0.004282	19.142853	0.026426	0.004282	19.142853	1	True	8
5	RandomForestEntr	0.9384	0.949950	accuracy	0.157505	0.075409	1.902846	0.157505	0.075409	1.902846	1	True	7
6	ExtraTreesGini	0.9360	0.946947	accuracy	0.204237	0.085555	1.144608	0.204237	0.085555	1.144608	1	True	9
7	ExtraTreesEntr	0.9358	0.942943	accuracy	0.224776	0.074932	1.114300	0.224776	0.074932	1.114300	1	True	10
8	RandomForestGini	0.9352	0.944945	accuracy	0.140813	0.075244	1.501600	0.140813	0.075244	1.501600	1	True	6
9	NeuralNetFastAI	0.9344	0.939940	accuracy	0.054227	0.012860	8.783859	0.054227	0.012860	8.783859	1	True	3
10	LightGBMXT	0.9320	0.945946	accuracy	0.251043	0.046257	4.116327	0.251043	0.046257	4.116327	1	True	4
11	NeuralNetTorch	0.9312	0.951952	accuracy	0.020438	0.008191	265.213656	0.020438	0.008191	265.213656	1	True	12
12	KNeighborsDist	0.2210	0.213213	accuracy	0.025489	0.013994	0.024563	0.025489	0.013994	0.024563	1	True	2
13	KNeighborsUnif	0.2180	0.223223	accuracy	0.015486	0.014760	0.049749	0.015486	0.014760	0.049749	1	True	1

Conclusion

In this quickstart tutorial we saw AutoGluon’s basic fit and predict functionality using TabularDataset and TabularPredictor. AutoGluon simplifies the model training process by not requiring feature engineering or model hyperparameter tuning. Check out the in-depth tutorials to learn more about AutoGluon’s other features like customizing the training and prediction steps or extending AutoGluon with custom feature generators, models, or metrics.