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
Requirement already satisfied: pip in /home/ci/opt/venv/lib/python3.11/site-packages (25.1.1)
Requirement already satisfied: autogluon in /home/ci/autogluon/autogluon/src (1.3.1b20250520)
Requirement already satisfied: autogluon.core==1.3.1b20250520 in /home/ci/autogluon/core/src (from autogluon.core[all]==1.3.1b20250520->autogluon) (1.3.1b20250520)
Requirement already satisfied: autogluon.features==1.3.1b20250520 in /home/ci/autogluon/features/src (from autogluon) (1.3.1b20250520)
Requirement already satisfied: autogluon.tabular==1.3.1b20250520 in /home/ci/autogluon/tabular/src (from autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.3.1b20250520)
Requirement already satisfied: autogluon.multimodal==1.3.1b20250520 in /home/ci/autogluon/multimodal/src (from autogluon) (1.3.1b20250520)
Requirement already satisfied: autogluon.timeseries==1.3.1b20250520 in /home/ci/autogluon/timeseries/src (from autogluon.timeseries[all]==1.3.1b20250520->autogluon) (1.3.1b20250520)
Requirement already satisfied: numpy<2.3.0,>=1.25.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (2.1.3)
Requirement already satisfied: scipy<1.16,>=1.5.4 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (1.15.3)
Requirement already satisfied: scikit-learn<1.7.0,>=1.4.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (1.6.1)
Requirement already satisfied: networkx<4,>=3.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (3.4.2)
Requirement already satisfied: pandas<2.3.0,>=2.0.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (2.2.3)
Requirement already satisfied: tqdm<5,>=4.38 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (4.67.1)
Requirement already satisfied: requests in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (2.32.3)
Requirement already satisfied: matplotlib<3.11,>=3.7.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (3.10.3)
Requirement already satisfied: boto3<2,>=1.10 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (1.38.19)
Requirement already satisfied: autogluon.common==1.3.1b20250520 in /home/ci/autogluon/common/src (from autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (1.3.1b20250520)
Requirement already satisfied: psutil<7.1.0,>=5.7.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.common==1.3.1b20250520->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (7.0.0)
Requirement already satisfied: hyperopt<0.2.8,>=0.2.7 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core[all]==1.3.1b20250520->autogluon) (0.2.7)
Requirement already satisfied: pyarrow>=15.0.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.core[all]==1.3.1b20250520->autogluon) (20.0.0)
Requirement already satisfied: ray<2.45,>=2.10.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (2.44.1)
Requirement already satisfied: Pillow<12,>=10.0.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (11.2.1)
Requirement already satisfied: torch<2.7,>=2.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (2.6.0)
Requirement already satisfied: lightning<2.7,>=2.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (2.5.1.post0)
Requirement already satisfied: transformers<4.50,>=4.38.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from transformers[sentencepiece]<4.50,>=4.38.0->autogluon.multimodal==1.3.1b20250520->autogluon) (4.49.0)
Requirement already satisfied: accelerate<2.0,>=0.34.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (1.7.0)
Requirement already satisfied: jsonschema<4.24,>=4.18 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (4.23.0)
Requirement already satisfied: seqeval<1.3.0,>=1.2.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (1.2.2)
Requirement already satisfied: evaluate<0.5.0,>=0.4.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (0.4.3)
Requirement already satisfied: timm<1.0.7,>=0.9.5 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (1.0.3)
Requirement already satisfied: torchvision<0.22.0,>=0.16.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (0.21.0)
Requirement already satisfied: scikit-image<0.26.0,>=0.19.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (0.25.2)
Requirement already satisfied: text-unidecode<1.4,>=1.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (1.3)
Requirement already satisfied: torchmetrics<1.8,>=1.2.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (1.7.1)
Requirement already satisfied: omegaconf<2.4.0,>=2.1.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (2.3.0)
Requirement already satisfied: pytorch-metric-learning<2.9,>=1.3.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (2.8.1)
Requirement already satisfied: nlpaug<1.2.0,>=1.1.10 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (1.1.11)
Requirement already satisfied: nltk<3.9,>=3.4.5 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (3.8.1)
Requirement already satisfied: openmim<0.4.0,>=0.3.7 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (0.3.9)
Requirement already satisfied: defusedxml<0.7.2,>=0.7.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (0.7.1)
Requirement already satisfied: jinja2<3.2,>=3.0.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (3.1.6)
Requirement already satisfied: tensorboard<3,>=2.9 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (2.19.0)
Requirement already satisfied: pytesseract<0.4,>=0.3.9 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (0.3.13)
Requirement already satisfied: nvidia-ml-py3<8.0,>=7.352.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (7.352.0)
Requirement already satisfied: pdf2image<1.19,>=1.17.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.multimodal==1.3.1b20250520->autogluon) (1.17.0)
Requirement already satisfied: einops<0.9,>=0.7 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.tabular[all]==1.3.1b20250520->autogluon) (0.8.1)
Requirement already satisfied: spacy<3.9 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.tabular[all]==1.3.1b20250520->autogluon) (3.8.6)
Requirement already satisfied: xgboost<3.1,>=2.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.tabular[all]==1.3.1b20250520->autogluon) (3.0.1)
Requirement already satisfied: catboost<1.3,>=1.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.2.8)
Requirement already satisfied: huggingface_hub[torch] in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.tabular[all]==1.3.1b20250520->autogluon) (0.31.4)
Requirement already satisfied: lightgbm<4.7,>=4.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.tabular[all]==1.3.1b20250520->autogluon) (4.6.0)
Requirement already satisfied: fastai<2.9,>=2.3.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.tabular[all]==1.3.1b20250520->autogluon) (2.8.1)
Requirement already satisfied: joblib<2,>=1.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (1.5.0)
Requirement already satisfied: pytorch_lightning in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (2.5.1.post0)
Requirement already satisfied: gluonts<0.17,>=0.15.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.16.1)
Requirement already satisfied: statsforecast<2.0.2,>=1.7.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (2.0.1)
Requirement already satisfied: mlforecast<0.14,>0.13 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.13.6)
Requirement already satisfied: utilsforecast<0.2.11,>=0.2.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.2.10)
Requirement already satisfied: coreforecast<0.0.16,>=0.0.12 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.0.15)
Requirement already satisfied: fugue>=0.9.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.9.1)
Requirement already satisfied: orjson~=3.9 in /home/ci/opt/venv/lib/python3.11/site-packages (from autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (3.10.18)
Requirement already satisfied: packaging>=20.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from accelerate<2.0,>=0.34.0->autogluon.multimodal==1.3.1b20250520->autogluon) (24.2)
Requirement already satisfied: pyyaml in /home/ci/opt/venv/lib/python3.11/site-packages (from accelerate<2.0,>=0.34.0->autogluon.multimodal==1.3.1b20250520->autogluon) (6.0.2)
Requirement already satisfied: safetensors>=0.4.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from accelerate<2.0,>=0.34.0->autogluon.multimodal==1.3.1b20250520->autogluon) (0.5.3)
Requirement already satisfied: botocore<1.39.0,>=1.38.19 in /home/ci/opt/venv/lib/python3.11/site-packages (from boto3<2,>=1.10->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (1.38.19)
Requirement already satisfied: jmespath<2.0.0,>=0.7.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from boto3<2,>=1.10->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (1.0.1)
Requirement already satisfied: s3transfer<0.13.0,>=0.12.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from boto3<2,>=1.10->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (0.12.0)
Requirement already satisfied: python-dateutil<3.0.0,>=2.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from botocore<1.39.0,>=1.38.19->boto3<2,>=1.10->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (2.9.0.post0)
Requirement already satisfied: urllib3!=2.2.0,<3,>=1.25.4 in /home/ci/opt/venv/lib/python3.11/site-packages (from botocore<1.39.0,>=1.38.19->boto3<2,>=1.10->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (2.4.0)
Requirement already satisfied: graphviz in /home/ci/opt/venv/lib/python3.11/site-packages (from catboost<1.3,>=1.2->autogluon.tabular[all]==1.3.1b20250520->autogluon) (0.20.3)
Requirement already satisfied: plotly in /home/ci/opt/venv/lib/python3.11/site-packages (from catboost<1.3,>=1.2->autogluon.tabular[all]==1.3.1b20250520->autogluon) (6.1.0)
Requirement already satisfied: six in /home/ci/opt/venv/lib/python3.11/site-packages (from catboost<1.3,>=1.2->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.17.0)
Requirement already satisfied: datasets>=2.0.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.1b20250520->autogluon) (2.14.4)
Requirement already satisfied: dill in /home/ci/opt/venv/lib/python3.11/site-packages (from evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.1b20250520->autogluon) (0.3.7)
Requirement already satisfied: xxhash in /home/ci/opt/venv/lib/python3.11/site-packages (from evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.1b20250520->autogluon) (3.5.0)
Requirement already satisfied: multiprocess in /home/ci/opt/venv/lib/python3.11/site-packages (from evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.1b20250520->autogluon) (0.70.15)
Requirement already satisfied: fsspec>=2021.05.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from fsspec[http]>=2021.05.0->evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.1b20250520->autogluon) (2025.3.2)
Requirement already satisfied: pip in /home/ci/opt/venv/lib/python3.11/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.1b20250520->autogluon) (25.1.1)
Requirement already satisfied: fastdownload<2,>=0.0.5 in /home/ci/opt/venv/lib/python3.11/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.1b20250520->autogluon) (0.0.7)
Requirement already satisfied: fastcore<1.9,>=1.8.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.8.2)
Requirement already satisfied: fasttransform>=0.0.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.1b20250520->autogluon) (0.0.2)
Requirement already satisfied: fastprogress>=0.2.4 in /home/ci/opt/venv/lib/python3.11/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.0.3)
Requirement already satisfied: plum-dispatch in /home/ci/opt/venv/lib/python3.11/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.1b20250520->autogluon) (2.5.7)
Requirement already satisfied: cloudpickle in /home/ci/opt/venv/lib/python3.11/site-packages (from fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.1b20250520->autogluon) (3.1.1)
Requirement already satisfied: pydantic<3,>=1.7 in /home/ci/opt/venv/lib/python3.11/site-packages (from gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (2.11.4)
Requirement already satisfied: toolz~=0.10 in /home/ci/opt/venv/lib/python3.11/site-packages (from gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.12.1)
Requirement already satisfied: typing-extensions~=4.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (4.13.2)
Requirement already satisfied: future in /home/ci/opt/venv/lib/python3.11/site-packages (from hyperopt<0.2.8,>=0.2.7->autogluon.core[all]==1.3.1b20250520->autogluon) (1.0.0)
Requirement already satisfied: py4j in /home/ci/opt/venv/lib/python3.11/site-packages (from hyperopt<0.2.8,>=0.2.7->autogluon.core[all]==1.3.1b20250520->autogluon) (0.10.9.9)
Requirement already satisfied: MarkupSafe>=2.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from jinja2<3.2,>=3.0.3->autogluon.multimodal==1.3.1b20250520->autogluon) (3.0.2)
Requirement already satisfied: attrs>=22.2.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from jsonschema<4.24,>=4.18->autogluon.multimodal==1.3.1b20250520->autogluon) (25.3.0)
Requirement already satisfied: jsonschema-specifications>=2023.03.6 in /home/ci/opt/venv/lib/python3.11/site-packages (from jsonschema<4.24,>=4.18->autogluon.multimodal==1.3.1b20250520->autogluon) (2025.4.1)
Requirement already satisfied: referencing>=0.28.4 in /home/ci/opt/venv/lib/python3.11/site-packages (from jsonschema<4.24,>=4.18->autogluon.multimodal==1.3.1b20250520->autogluon) (0.36.2)
Requirement already satisfied: rpds-py>=0.7.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from jsonschema<4.24,>=4.18->autogluon.multimodal==1.3.1b20250520->autogluon) (0.25.0)
Requirement already satisfied: lightning-utilities<2.0,>=0.10.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from lightning<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (0.14.3)
Requirement already satisfied: aiohttp!=4.0.0a0,!=4.0.0a1 in /home/ci/opt/venv/lib/python3.11/site-packages (from fsspec[http]>=2021.05.0->evaluate<0.5.0,>=0.4.0->autogluon.multimodal==1.3.1b20250520->autogluon) (3.11.18)
Requirement already satisfied: setuptools in /home/ci/opt/venv/lib/python3.11/site-packages (from lightning-utilities<2.0,>=0.10.0->lightning<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (65.5.0)
Requirement already satisfied: contourpy>=1.0.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from matplotlib<3.11,>=3.7.0->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (1.3.2)
Requirement already satisfied: cycler>=0.10 in /home/ci/opt/venv/lib/python3.11/site-packages (from matplotlib<3.11,>=3.7.0->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (0.12.1)
Requirement already satisfied: fonttools>=4.22.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from matplotlib<3.11,>=3.7.0->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (4.58.0)
Requirement already satisfied: kiwisolver>=1.3.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from matplotlib<3.11,>=3.7.0->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (1.4.8)
Requirement already satisfied: pyparsing>=2.3.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from matplotlib<3.11,>=3.7.0->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (3.2.3)
Requirement already satisfied: numba in /home/ci/opt/venv/lib/python3.11/site-packages (from mlforecast<0.14,>0.13->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.61.2)
Requirement already satisfied: optuna in /home/ci/opt/venv/lib/python3.11/site-packages (from mlforecast<0.14,>0.13->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (4.3.0)
Requirement already satisfied: window-ops in /home/ci/opt/venv/lib/python3.11/site-packages (from mlforecast<0.14,>0.13->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.0.15)
Requirement already satisfied: gdown>=4.0.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from nlpaug<1.2.0,>=1.1.10->autogluon.multimodal==1.3.1b20250520->autogluon) (5.2.0)
Requirement already satisfied: click in /home/ci/opt/venv/lib/python3.11/site-packages (from nltk<3.9,>=3.4.5->autogluon.multimodal==1.3.1b20250520->autogluon) (8.1.8)
Requirement already satisfied: regex>=2021.8.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from nltk<3.9,>=3.4.5->autogluon.multimodal==1.3.1b20250520->autogluon) (2024.11.6)
Requirement already satisfied: antlr4-python3-runtime==4.9.* in /home/ci/opt/venv/lib/python3.11/site-packages (from omegaconf<2.4.0,>=2.1.1->autogluon.multimodal==1.3.1b20250520->autogluon) (4.9.3)
Requirement already satisfied: colorama in /home/ci/opt/venv/lib/python3.11/site-packages (from openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (0.4.4)
Requirement already satisfied: model-index in /home/ci/opt/venv/lib/python3.11/site-packages (from openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (0.1.11)
Requirement already satisfied: opendatalab in /home/ci/opt/venv/lib/python3.11/site-packages (from openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (0.0.10)
Requirement already satisfied: rich in /home/ci/opt/venv/lib/python3.11/site-packages (from openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (14.0.0)
Requirement already satisfied: tabulate in /home/ci/opt/venv/lib/python3.11/site-packages (from openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (0.9.0)
Requirement already satisfied: pytz>=2020.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from pandas<2.3.0,>=2.0.0->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (2025.2)
Requirement already satisfied: tzdata>=2022.7 in /home/ci/opt/venv/lib/python3.11/site-packages (from pandas<2.3.0,>=2.0.0->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (2025.2)
Requirement already satisfied: annotated-types>=0.6.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from pydantic<3,>=1.7->gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.7.0)
Requirement already satisfied: pydantic-core==2.33.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from pydantic<3,>=1.7->gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (2.33.2)
Requirement already satisfied: typing-inspection>=0.4.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from pydantic<3,>=1.7->gluonts<0.17,>=0.15.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.4.0)
Requirement already satisfied: filelock in /home/ci/opt/venv/lib/python3.11/site-packages (from ray<2.45,>=2.10.0->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (3.18.0)
Requirement already satisfied: msgpack<2.0.0,>=1.0.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from ray<2.45,>=2.10.0->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (1.1.0)
Requirement already satisfied: protobuf!=3.19.5,>=3.15.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from ray<2.45,>=2.10.0->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (3.20.2)
Requirement already satisfied: aiosignal in /home/ci/opt/venv/lib/python3.11/site-packages (from ray<2.45,>=2.10.0->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (1.3.2)
Requirement already satisfied: frozenlist in /home/ci/opt/venv/lib/python3.11/site-packages (from ray<2.45,>=2.10.0->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (1.6.0)
Requirement already satisfied: aiohttp_cors in /home/ci/opt/venv/lib/python3.11/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (0.8.1)
Requirement already satisfied: colorful in /home/ci/opt/venv/lib/python3.11/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (0.5.6)
Requirement already satisfied: py-spy>=0.2.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (0.4.0)
Requirement already satisfied: grpcio>=1.42.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (1.71.0)
Requirement already satisfied: opencensus in /home/ci/opt/venv/lib/python3.11/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (0.11.4)
Requirement already satisfied: prometheus_client>=0.7.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (0.22.0)
Requirement already satisfied: smart_open in /home/ci/opt/venv/lib/python3.11/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (7.1.0)
Requirement already satisfied: virtualenv!=20.21.1,>=20.0.24 in /home/ci/opt/venv/lib/python3.11/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (20.31.2)
Requirement already satisfied: tensorboardX>=1.9 in /home/ci/opt/venv/lib/python3.11/site-packages (from ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (2.6.2.2)
Requirement already satisfied: charset-normalizer<4,>=2 in /home/ci/opt/venv/lib/python3.11/site-packages (from requests->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (3.4.2)
Requirement already satisfied: idna<4,>=2.5 in /home/ci/opt/venv/lib/python3.11/site-packages (from requests->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (3.10)
Requirement already satisfied: certifi>=2017.4.17 in /home/ci/opt/venv/lib/python3.11/site-packages (from requests->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (2025.4.26)
Requirement already satisfied: imageio!=2.35.0,>=2.33 in /home/ci/opt/venv/lib/python3.11/site-packages (from scikit-image<0.26.0,>=0.19.1->autogluon.multimodal==1.3.1b20250520->autogluon) (2.37.0)
Requirement already satisfied: tifffile>=2022.8.12 in /home/ci/opt/venv/lib/python3.11/site-packages (from scikit-image<0.26.0,>=0.19.1->autogluon.multimodal==1.3.1b20250520->autogluon) (2025.5.10)
Requirement already satisfied: lazy-loader>=0.4 in /home/ci/opt/venv/lib/python3.11/site-packages (from scikit-image<0.26.0,>=0.19.1->autogluon.multimodal==1.3.1b20250520->autogluon) (0.4)
Requirement already satisfied: threadpoolctl>=3.1.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from scikit-learn<1.7.0,>=1.4.0->autogluon.core==1.3.1b20250520->autogluon.core[all]==1.3.1b20250520->autogluon) (3.6.0)
Requirement already satisfied: spacy-legacy<3.1.0,>=3.0.11 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (3.0.12)
Requirement already satisfied: spacy-loggers<2.0.0,>=1.0.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.0.5)
Requirement already satisfied: murmurhash<1.1.0,>=0.28.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.0.12)
Requirement already satisfied: cymem<2.1.0,>=2.0.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (2.0.11)
Requirement already satisfied: preshed<3.1.0,>=3.0.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (3.0.9)
Requirement already satisfied: thinc<8.4.0,>=8.3.4 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (8.3.6)
Requirement already satisfied: wasabi<1.2.0,>=0.9.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.1.3)
Requirement already satisfied: srsly<3.0.0,>=2.4.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (2.5.1)
Requirement already satisfied: catalogue<2.1.0,>=2.0.6 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (2.0.10)
Requirement already satisfied: weasel<0.5.0,>=0.1.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (0.4.1)
Requirement already satisfied: typer<1.0.0,>=0.3.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (0.15.4)
Requirement already satisfied: langcodes<4.0.0,>=3.2.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (3.5.0)
Requirement already satisfied: language-data>=1.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from langcodes<4.0.0,>=3.2.0->spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.3.0)
Requirement already satisfied: statsmodels>=0.13.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from statsforecast<2.0.2,>=1.7.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.14.4)
Requirement already satisfied: absl-py>=0.4 in /home/ci/opt/venv/lib/python3.11/site-packages (from tensorboard<3,>=2.9->autogluon.multimodal==1.3.1b20250520->autogluon) (2.2.2)
Requirement already satisfied: markdown>=2.6.8 in /home/ci/opt/venv/lib/python3.11/site-packages (from tensorboard<3,>=2.9->autogluon.multimodal==1.3.1b20250520->autogluon) (3.8)
Requirement already satisfied: tensorboard-data-server<0.8.0,>=0.7.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from tensorboard<3,>=2.9->autogluon.multimodal==1.3.1b20250520->autogluon) (0.7.2)
Requirement already satisfied: werkzeug>=1.0.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from tensorboard<3,>=2.9->autogluon.multimodal==1.3.1b20250520->autogluon) (3.1.3)
Requirement already satisfied: blis<1.4.0,>=1.3.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from thinc<8.4.0,>=8.3.4->spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.3.0)
Requirement already satisfied: confection<1.0.0,>=0.0.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from thinc<8.4.0,>=8.3.4->spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (0.1.5)
Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.4.127 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (12.4.127)
Requirement already satisfied: nvidia-cuda-runtime-cu12==12.4.127 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (12.4.127)
Requirement already satisfied: nvidia-cuda-cupti-cu12==12.4.127 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (12.4.127)
Requirement already satisfied: nvidia-cudnn-cu12==9.1.0.70 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (9.1.0.70)
Requirement already satisfied: nvidia-cublas-cu12==12.4.5.8 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (12.4.5.8)
Requirement already satisfied: nvidia-cufft-cu12==11.2.1.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (11.2.1.3)
Requirement already satisfied: nvidia-curand-cu12==10.3.5.147 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (10.3.5.147)
Requirement already satisfied: nvidia-cusolver-cu12==11.6.1.9 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (11.6.1.9)
Requirement already satisfied: nvidia-cusparse-cu12==12.3.1.170 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (12.3.1.170)
Requirement already satisfied: nvidia-cusparselt-cu12==0.6.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (0.6.2)
Requirement already satisfied: nvidia-nccl-cu12==2.21.5 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (2.21.5)
Requirement already satisfied: nvidia-nvtx-cu12==12.4.127 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (12.4.127)
Requirement already satisfied: nvidia-nvjitlink-cu12==12.4.127 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (12.4.127)
Requirement already satisfied: triton==3.2.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (3.2.0)
Requirement already satisfied: sympy==1.13.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (1.13.1)
Requirement already satisfied: mpmath<1.4,>=1.1.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from sympy==1.13.1->torch<2.7,>=2.2->autogluon.multimodal==1.3.1b20250520->autogluon) (1.3.0)
Requirement already satisfied: tokenizers<0.22,>=0.21 in /home/ci/opt/venv/lib/python3.11/site-packages (from transformers<4.50,>=4.38.0->transformers[sentencepiece]<4.50,>=4.38.0->autogluon.multimodal==1.3.1b20250520->autogluon) (0.21.1)
Requirement already satisfied: sentencepiece!=0.1.92,>=0.1.91 in /home/ci/opt/venv/lib/python3.11/site-packages (from transformers[sentencepiece]<4.50,>=4.38.0->autogluon.multimodal==1.3.1b20250520->autogluon) (0.2.0)
Requirement already satisfied: shellingham>=1.3.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from typer<1.0.0,>=0.3.0->spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.5.4)
Requirement already satisfied: cloudpathlib<1.0.0,>=0.7.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from weasel<0.5.0,>=0.1.0->spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (0.21.1)
Requirement already satisfied: wrapt in /home/ci/opt/venv/lib/python3.11/site-packages (from smart_open->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (1.17.2)
Requirement already satisfied: aiohappyeyeballs>=2.3.0 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (2.6.1)
Requirement already satisfied: multidict<7.0,>=4.5 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (6.4.4)
Requirement already satisfied: propcache>=0.2.0 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (0.3.1)
Requirement already satisfied: yarl<2.0,>=1.17.0 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (1.20.0)
Requirement already satisfied: triad>=0.9.7 in /home/ci/opt/venv/lib/python3.11/site-packages (from fugue>=0.9.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.9.8)
Requirement already satisfied: adagio>=0.2.4 in /home/ci/opt/venv/lib/python3.11/site-packages (from fugue>=0.9.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.2.6)
Requirement already satisfied: beautifulsoup4 in /home/ci/opt/venv/lib/python3.11/site-packages (from gdown>=4.0.0->nlpaug<1.2.0,>=1.1.10->autogluon.multimodal==1.3.1b20250520->autogluon) (4.13.4)
Requirement already satisfied: marisa-trie>=1.1.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from language-data>=1.2->langcodes<4.0.0,>=3.2.0->spacy<3.9->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.2.1)
Requirement already satisfied: llvmlite<0.45,>=0.44.0dev0 in /home/ci/opt/venv/lib/python3.11/site-packages (from numba->mlforecast<0.14,>0.13->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (0.44.0)
Requirement already satisfied: markdown-it-py>=2.2.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from rich->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (3.0.0)
Requirement already satisfied: pygments<3.0.0,>=2.13.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from rich->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (2.19.1)
Requirement already satisfied: mdurl~=0.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from markdown-it-py>=2.2.0->rich->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (0.1.2)
Requirement already satisfied: patsy>=0.5.6 in /home/ci/opt/venv/lib/python3.11/site-packages (from statsmodels>=0.13.2->statsforecast<2.0.2,>=1.7.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (1.0.1)
Requirement already satisfied: fs in /home/ci/opt/venv/lib/python3.11/site-packages (from triad>=0.9.7->fugue>=0.9.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (2.4.16)
Requirement already satisfied: distlib<1,>=0.3.7 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (0.3.9)
Requirement already satisfied: platformdirs<5,>=3.9.1 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (4.3.8)
Requirement already satisfied: soupsieve>1.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from beautifulsoup4->gdown>=4.0.0->nlpaug<1.2.0,>=1.1.10->autogluon.multimodal==1.3.1b20250520->autogluon) (2.7)
Requirement already satisfied: appdirs~=1.4.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from fs->triad>=0.9.7->fugue>=0.9.0->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (1.4.4)
Requirement already satisfied: ordered-set in /home/ci/opt/venv/lib/python3.11/site-packages (from model-index->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (4.1.0)
Requirement already satisfied: opencensus-context>=0.1.3 in /home/ci/opt/venv/lib/python3.11/site-packages (from opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (0.1.3)
Requirement already satisfied: google-api-core<3.0.0,>=1.0.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from opencensus->ray[default,tune]<2.45,>=2.10.0; extra == "all"->autogluon.core[all]==1.3.1b20250520->autogluon) (2.24.2)
Requirement already satisfied: googleapis-common-protos<2.0.0,>=1.56.2 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (1.70.0)
Requirement already satisfied: proto-plus<2.0.0,>=1.22.3 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (1.26.1)
Requirement already satisfied: google-auth<3.0.0,>=2.14.1 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (2.40.1)
Requirement already satisfied: cachetools<6.0,>=2.0.0 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (5.5.2)
Requirement already satisfied: pyasn1-modules>=0.2.1 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (0.4.2)
Requirement already satisfied: rsa<5,>=3.1.4 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (4.9.1)
Requirement already satisfied: pyasn1>=0.1.3 in /home/ci/opt/venv/lib/python3.11/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.1b20250520->autogluon) (0.6.1)
Requirement already satisfied: pycryptodome in /home/ci/opt/venv/lib/python3.11/site-packages (from opendatalab->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (3.23.0)
Requirement already satisfied: openxlab in /home/ci/opt/venv/lib/python3.11/site-packages (from opendatalab->openmim<0.4.0,>=0.3.7->autogluon.multimodal==1.3.1b20250520->autogluon) (0.0.11)
Requirement already satisfied: alembic>=1.5.0 in /home/ci/opt/venv/lib/python3.11/site-packages (from optuna->mlforecast<0.14,>0.13->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (1.15.2)
Requirement already satisfied: colorlog in /home/ci/opt/venv/lib/python3.11/site-packages (from optuna->mlforecast<0.14,>0.13->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (6.9.0)
Requirement already satisfied: sqlalchemy>=1.4.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from optuna->mlforecast<0.14,>0.13->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (2.0.41)
Requirement already satisfied: Mako in /home/ci/opt/venv/lib/python3.11/site-packages (from alembic>=1.5.0->optuna->mlforecast<0.14,>0.13->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (1.3.10)
Requirement already satisfied: greenlet>=1 in /home/ci/opt/venv/lib/python3.11/site-packages (from sqlalchemy>=1.4.2->optuna->mlforecast<0.14,>0.13->autogluon.timeseries==1.3.1b20250520->autogluon.timeseries[all]==1.3.1b20250520->autogluon) (3.2.2)
Requirement already satisfied: narwhals>=1.15.1 in /home/ci/opt/venv/lib/python3.11/site-packages (from plotly->catboost<1.3,>=1.2->autogluon.tabular[all]==1.3.1b20250520->autogluon) (1.40.0)
Requirement already satisfied: beartype>=0.16.2 in /home/ci/opt/venv/lib/python3.11/site-packages (from plum-dispatch->fastai<2.9,>=2.3.1->autogluon.tabular[all]==1.3.1b20250520->autogluon) (0.20.2)
Requirement already satisfied: PySocks!=1.5.7,>=1.5.6 in /home/ci/opt/venv/lib/python3.11/site-packages (from requests[socks]->gdown>=4.0.0->nlpaug<1.2.0,>=1.1.10->autogluon.multimodal==1.3.1b20250520->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
No path specified. Models will be saved in: "AutogluonModels/ag-20250520_120042"
Verbosity: 2 (Standard Logging)
=================== System Info ===================
AutoGluon Version: 1.3.1b20250520
Python Version: 3.11.9
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.69 GB / 30.95 GB (92.7%)
Disk Space Avail: 205.91 GB / 255.99 GB (80.4%)
===================================================
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.
Beginning AutoGluon training ...
AutoGluon will save models to "/home/ci/autogluon/docs/tutorials/tabular/AutogluonModels/ag-20250520_120042"
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: 29380.57 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.11s ...
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.02s = Validation runtime
Fitting model: KNeighborsDist ...
0.2132 = Validation score (accuracy)
0.03s = Training runtime
0.01s = Validation runtime
Fitting model: NeuralNetFastAI ...
0.9459 = Validation score (accuracy)
9.88s = Training runtime
0.01s = Validation runtime
Fitting model: LightGBMXT ...
0.9459 = Validation score (accuracy)
4.32s = Training runtime
0.05s = Validation runtime
Fitting model: LightGBM ...
0.956 = Validation score (accuracy)
3.79s = Training runtime
0.03s = Validation runtime
Fitting model: RandomForestGini ...
0.9449 = Validation score (accuracy)
1.43s = Training runtime
0.08s = Validation runtime
Fitting model: RandomForestEntr ...
0.9499 = Validation score (accuracy)
1.82s = Training runtime
0.08s = Validation runtime
Fitting model: CatBoost ...
0.956 = Validation score (accuracy)
20.15s = Training runtime
0.0s = Validation runtime
Fitting model: ExtraTreesGini ...
0.9469 = Validation score (accuracy)
1.14s = Training runtime
0.08s = Validation runtime
Fitting model: ExtraTreesEntr ...
0.9429 = Validation score (accuracy)
1.14s = Training runtime
0.09s = Validation runtime
Fitting model: XGBoost ...
0.957 = Validation score (accuracy)
5.21s = Training runtime
0.07s = Validation runtime
Fitting model: NeuralNetTorch ...
0.9409 = Validation score (accuracy)
52.15s = Training runtime
0.01s = Validation runtime
Fitting model: LightGBMLarge ...
0.9499 = Validation score (accuracy)
8.05s = Training runtime
0.07s = Validation runtime
Fitting model: WeightedEnsemble_L2 ...
Ensemble Weights: {'RandomForestEntr': 0.429, 'NeuralNetFastAI': 0.286, 'ExtraTreesGini': 0.143, 'XGBoost': 0.143}
0.965 = Validation score (accuracy)
0.12s = Training runtime
0.0s = Validation runtime
AutoGluon training complete, total runtime = 111.3s ... Best model: WeightedEnsemble_L2 | Estimated inference throughput: 4246.1 rows/s (999 batch size)
TabularPredictor saved. To load, use: predictor = TabularPredictor.load("/home/ci/autogluon/docs/tutorials/tabular/AutogluonModels/ag-20250520_120042")
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()
---------------------------------------------------------------------------
HTTPError Traceback (most recent call last)
Cell In[6], line 1
----> 1 test_data = TabularDataset(f'{data_url}test.csv')
3 y_pred = predictor.predict(test_data.drop(columns=[label]))
4 y_pred.head()
File ~/autogluon/common/src/autogluon/common/dataset.py:35, in TabularDataset.__new__(cls, data, **kwargs)
33 def __new__(cls, data, **kwargs) -> pd.DataFrame:
34 if isinstance(data, str):
---> 35 data = load_pd.load(data)
36 return pd.DataFrame(data, **kwargs)
File ~/autogluon/common/src/autogluon/common/loaders/load_pd.py:143, in load(path, delimiter, encoding, columns_to_keep, dtype, header, names, format, nrows, skiprows, usecols, low_memory, converters, filters, sample_count, worker_count, multiprocessing_method)
141 column_count_full = len(df.columns)
142 elif format == "csv":
--> 143 df = pd.read_csv(
144 path,
145 converters=converters,
146 delimiter=delimiter,
147 encoding=encoding,
148 header=header,
149 names=names,
150 dtype=dtype,
151 low_memory=low_memory,
152 nrows=nrows,
153 skiprows=skiprows,
154 usecols=usecols,
155 )
156 column_count_full = len(list(df.columns.values))
157 if columns_to_keep is not None:
File ~/opt/venv/lib/python3.11/site-packages/pandas/io/parsers/readers.py:1026, in read_csv(filepath_or_buffer, sep, delimiter, header, names, index_col, usecols, dtype, engine, converters, true_values, false_values, skipinitialspace, skiprows, skipfooter, nrows, na_values, keep_default_na, na_filter, verbose, skip_blank_lines, parse_dates, infer_datetime_format, keep_date_col, date_parser, date_format, dayfirst, cache_dates, iterator, chunksize, compression, thousands, decimal, lineterminator, quotechar, quoting, doublequote, escapechar, comment, encoding, encoding_errors, dialect, on_bad_lines, delim_whitespace, low_memory, memory_map, float_precision, storage_options, dtype_backend)
1013 kwds_defaults = _refine_defaults_read(
1014 dialect,
1015 delimiter,
(...)
1022 dtype_backend=dtype_backend,
1023 )
1024 kwds.update(kwds_defaults)
-> 1026 return _read(filepath_or_buffer, kwds)
File ~/opt/venv/lib/python3.11/site-packages/pandas/io/parsers/readers.py:620, in _read(filepath_or_buffer, kwds)
617 _validate_names(kwds.get("names", None))
619 # Create the parser.
--> 620 parser = TextFileReader(filepath_or_buffer, **kwds)
622 if chunksize or iterator:
623 return parser
File ~/opt/venv/lib/python3.11/site-packages/pandas/io/parsers/readers.py:1620, in TextFileReader.__init__(self, f, engine, **kwds)
1617 self.options["has_index_names"] = kwds["has_index_names"]
1619 self.handles: IOHandles | None = None
-> 1620 self._engine = self._make_engine(f, self.engine)
File ~/opt/venv/lib/python3.11/site-packages/pandas/io/parsers/readers.py:1880, in TextFileReader._make_engine(self, f, engine)
1878 if "b" not in mode:
1879 mode += "b"
-> 1880 self.handles = get_handle(
1881 f,
1882 mode,
1883 encoding=self.options.get("encoding", None),
1884 compression=self.options.get("compression", None),
1885 memory_map=self.options.get("memory_map", False),
1886 is_text=is_text,
1887 errors=self.options.get("encoding_errors", "strict"),
1888 storage_options=self.options.get("storage_options", None),
1889 )
1890 assert self.handles is not None
1891 f = self.handles.handle
File ~/opt/venv/lib/python3.11/site-packages/pandas/io/common.py:728, in get_handle(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)
725 codecs.lookup_error(errors)
727 # open URLs
--> 728 ioargs = _get_filepath_or_buffer(
729 path_or_buf,
730 encoding=encoding,
731 compression=compression,
732 mode=mode,
733 storage_options=storage_options,
734 )
736 handle = ioargs.filepath_or_buffer
737 handles: list[BaseBuffer]
File ~/opt/venv/lib/python3.11/site-packages/pandas/io/common.py:384, in _get_filepath_or_buffer(filepath_or_buffer, encoding, compression, mode, storage_options)
382 # assuming storage_options is to be interpreted as headers
383 req_info = urllib.request.Request(filepath_or_buffer, headers=storage_options)
--> 384 with urlopen(req_info) as req:
385 content_encoding = req.headers.get("Content-Encoding", None)
386 if content_encoding == "gzip":
387 # Override compression based on Content-Encoding header
File ~/opt/venv/lib/python3.11/site-packages/pandas/io/common.py:289, in urlopen(*args, **kwargs)
283 """
284 Lazy-import wrapper for stdlib urlopen, as that imports a big chunk of
285 the stdlib.
286 """
287 import urllib.request
--> 289 return urllib.request.urlopen(*args, **kwargs)
File /opt/conda/lib/python3.11/urllib/request.py:216, in urlopen(url, data, timeout, cafile, capath, cadefault, context)
214 else:
215 opener = _opener
--> 216 return opener.open(url, data, timeout)
File /opt/conda/lib/python3.11/urllib/request.py:525, in OpenerDirector.open(self, fullurl, data, timeout)
523 for processor in self.process_response.get(protocol, []):
524 meth = getattr(processor, meth_name)
--> 525 response = meth(req, response)
527 return response
File /opt/conda/lib/python3.11/urllib/request.py:634, in HTTPErrorProcessor.http_response(self, request, response)
631 # According to RFC 2616, "2xx" code indicates that the client's
632 # request was successfully received, understood, and accepted.
633 if not (200 <= code < 300):
--> 634 response = self.parent.error(
635 'http', request, response, code, msg, hdrs)
637 return response
File /opt/conda/lib/python3.11/urllib/request.py:563, in OpenerDirector.error(self, proto, *args)
561 if http_err:
562 args = (dict, 'default', 'http_error_default') + orig_args
--> 563 return self._call_chain(*args)
File /opt/conda/lib/python3.11/urllib/request.py:496, in OpenerDirector._call_chain(self, chain, kind, meth_name, *args)
494 for handler in handlers:
495 func = getattr(handler, meth_name)
--> 496 result = func(*args)
497 if result is not None:
498 return result
File /opt/conda/lib/python3.11/urllib/request.py:643, in HTTPDefaultErrorHandler.http_error_default(self, req, fp, code, msg, hdrs)
642 def http_error_default(self, req, fp, code, msg, hdrs):
--> 643 raise HTTPError(req.full_url, code, msg, hdrs, fp)
HTTPError: HTTP Error 429: Too Many Requests
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)
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)
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.