autogluon 0.5.2


AutoML for Image, Text, and Tabular Data

AutoML for Image, Text, and Tabular Data

Stars: 4843, Watchers: 4843, Forks: 641, Open Issues: 183

The awslabs/autogluon repo was created 3 years ago and was last updated 40 minutes ago.
The project is very popular with an impressive 4843 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 183 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.5.0

HuggingFace community-driven open-source library of datasets

thinc 8.1.1

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.8

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