This release is a pre-release and may not be stable for production use.
Fast and Accurate ML in 3 Lines of Code
Installation | Documentation | Release Notes
AutoGluon automates machine learning tasks enabling you to easily achieve strong predictive performance in your applications. With just a few lines of code, you can train and deploy high-accuracy machine learning and deep learning models on image, text, time series, and tabular data.
💾 Installation
AutoGluon is supported on Python 3.9 - 3.12 and is available on Linux, MacOS, and Windows.
You can install AutoGluon with:
pip install autogluon
Visit our Installation Guide for detailed instructions, including GPU support, Conda installs, and optional dependencies.
:zap: Quickstart
Build accurate end-to-end ML models in just 3 lines of code!
from autogluon.tabular import TabularPredictor
predictor = TabularPredictor(label="class").fit("train.csv")
predictions = predictor.predict("test.csv")
| AutoGluon Task | Quickstart | API |
|---|---|---|
| TabularPredictor |  |
 |
| MultiModalPredictor | |
|
| TimeSeriesPredictor | |
|
:mag: Resources
Hands-on Tutorials / Talks
Below is a curated list of recent tutorials and talks on AutoGluon. A comprehensive list is available here.
| Title | Format | Location | Date |
|---|---|---|---|
| :tv: AutoGluon: Towards No-Code Automated Machine Learning | Tutorial | AutoML 2024 | 2024/09/09 |
| :tv: AutoGluon 1.0: Shattering the AutoML Ceiling with Zero Lines of Code | Tutorial | AutoML 2023 | 2023/09/12 |
| :sound: AutoGluon: The Story | Podcast | The AutoML Podcast | 2023/09/05 |
| :tv: AutoGluon: AutoML for Tabular, Multimodal, and Time Series Data | Tutorial | PyData Berlin | 2023/06/20 |
| :tv: Solving Complex ML Problems in a few Lines of Code with AutoGluon | Tutorial | PyData Seattle | 2023/06/20 |
| :tv: The AutoML Revolution | Tutorial | Fall AutoML School 2022 | 2022/10/18 |
Scientific Publications
- AutoGluon-Tabular: Robust and Accurate AutoML for Structured Data (Arxiv, 2020) (BibTeX)
- Fast, Accurate, and Simple Models for Tabular Data via Augmented Distillation (NeurIPS, 2020) (BibTeX)
- Benchmarking Multimodal AutoML for Tabular Data with Text Fields (NeurIPS, 2021) (BibTeX)
- XTab: Cross-table Pretraining for Tabular Transformers (ICML, 2023)
- AutoGluon-TimeSeries: AutoML for Probabilistic Time Series Forecasting (AutoML Conf, 2023) (BibTeX)
- TabRepo: A Large Scale Repository of Tabular Model Evaluations and its AutoML Applications (Under Review, 2024)
Articles
- AutoGluon-TimeSeries: Every Time Series Forecasting Model In One Library (Towards Data Science, Jan 2024)
- AutoGluon for tabular data: 3 lines of code to achieve top 1% in Kaggle competitions (AWS Open Source Blog, Mar 2020)
- AutoGluon overview & example applications (Towards Data Science, Dec 2019)
Train/Deploy AutoGluon in the Cloud
- AutoGluon Cloud (Recommended)
- AutoGluon on SageMaker AutoPilot
- AutoGluon on Amazon SageMaker
- AutoGluon Deep Learning Containers (Security certified & maintained by the AutoGluon developers)
- AutoGluon Official Docker Container
- AutoGluon-Tabular on AWS Marketplace (Not maintained by us)
:pencil: Citing AutoGluon
If you use AutoGluon in a scientific publication, please refer to our citation guide.
:wave: How to get involved
We are actively accepting code contributions to the AutoGluon project. If you are interested in contributing to AutoGluon, please read the Contributing Guide to get started.
:classical_building: License
This library is licensed under the Apache 2.0 License.
Release files for autogluon 1.2.1b20241218
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| autogluon-1.2.1b20241218.tar.gz | 5.3 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| autogluon-1.2.1b20241218-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 15.2 kB
Release files / autogluon-1.2.1b20241218.tar.gz
| Download URL | autogluon-1.2.1b20241218.tar.gz |
|---|---|
| Size | 5.3 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
b2ae1c41a070cc3c4508f25f23192707db4f2331c8c3600c208bebb8950dd5a8
|
|
BLAKE2b-256 checksum How to use checksums |
40c14259c3bfb91c667d31ada7c13df620a9ed55aea185b0d1a70a2b264df8e2
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.8.18
|
Release files / autogluon-1.2.1b20241218-py3-none-any.whl
| Download URL | autogluon-1.2.1b20241218-py3-none-any.whl |
|---|---|
| Size | 9.9 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
ce808ffb315c70d914ed1813f2b9aa071f3a583bc64ca8f81fa359319115955e
|
|
BLAKE2b-256 checksum How to use checksums |
218ffdbb62be56111fe9f6387a42f2846a9c076e8b09b6edb1250ba164bfd54d
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.8.18
|
