autogluon 0.4.0


AutoML for Image, Text, and Tabular Data

AutoML for Image, Text, and Tabular Data

Stars: 4465, Watchers: 4465, Forks: 588, Open Issues: 153

The awslabs/autogluon repo was created 2 years ago and was last updated 7 hours ago.
The project is very popular with an impressive 4465 github stars!

How to Install autogluon

You can install autogluon using pip

pip install autogluon

or add it to a project with poetry

poetry add autogluon

Package Details

AutoGluon Community
GitHub Repo


  • Scientific/Engineering/Artificial Intelligence
  • Scientific/Engineering/Image Recognition
  • Scientific/Engineering/Information Analysis
  • Software Development
No  autogluon  pypi packages just yet.


A list of common autogluon errors.

Code Examples

Here are some autogluon code examples and snippets.

GitHub Issues

The autogluon package has 153 open issues on GitHub

  • Various HPO Cleanup / Fixes
  • NN_TORCH incorrectly implements use_batchnorm=True
  • Handle nan in categorical feature in text feature generator
  • NeuralNetFastAI runs extremely slow
  • How do I set confidence when I use test data to predict? Are there relevant parameters?
  • [WIP]ARM/M1 dependency upgrade and doc
  • Text Prediction only be handled in English? Do you support Chinese?
  • Add scikit-learn compatible API
  • TabularPredictor config helper
  • Consider integrating cleanlab - cleans noisy labels (enhancement)
  • [v0.4] Add documentation for HPO / Search Spaces
  • [v0.4] Refactor model valid features to be allow instead of exclude
  • Feature Tools integration
  • Included Model Types
  • No GPU is detected in the machine and we will not proceed to run TextPredictor because they will train too slowly with only CPU.

See more issues on GitHub

Related Packages & Articles


H2O, Fast Scalable Machine Learning, for python

datasets 2.2.1

HuggingFace community-driven open-source library of datasets

thinc 8.0.15

A refreshing functional take on deep learning, compatible with your favorite libraries

nlp 0.4.0

HuggingFace/NLP is an open library of NLP datasets.

ktrain 0.31.0

ktrain is a wrapper for TensorFlow Keras that makes deep learning and AI more accessible and easier to apply