autogluon  by autogluon

Automated machine learning for diverse data types

Created 6 years ago
10,087 stars

Top 5.1% on SourcePulse

GitHubView on GitHub
Project Summary

AutoGluon addresses the complexity of machine learning by automating model training and deployment for tabular, time series, and multimodal data. It targets engineers and researchers seeking to achieve high predictive performance with minimal code. The primary benefit is accelerated development and deployment of accurate ML solutions across diverse data types.

How It Works

AutoGluon automates the end-to-end ML pipeline, abstracting complex model selection, hyperparameter tuning, and ensembling. It leverages a suite of models and advanced techniques, including augmented distillation and foundation models, to achieve state-of-the-art accuracy. The framework supports tabular, time series, and multimodal data, offering specialized predictors for each, with presets like "best" to guide the automated process. This approach significantly reduces the expertise and time required for effective ML deployment.

Quick Start & Requirements

  • Installation: pip install autogluon
  • Prerequisites: Python 3.10-3.13. Supports Linux, macOS, and Windows. GPU support is available and detailed in the installation guide.
  • Links: Installation Guide (https://auto.gluon.ai/stable/install.html), Documentation (https://auto.gluon.ai/stable/index.html).
  • Quickstart Example: from autogluon.tabular import TabularPredictor; predictor = TabularPredictor(label="class").fit("train.csv", presets="best"); predictions = predictor.predict("test.csv")

Highlighted Details

  • Claims "Fast and Accurate ML in 3 Lines of Code."
  • Supports tabular, time series, and multimodal data with specialized predictors.
  • Developed by AWS AI, with extensive research publications and benchmarks available.
  • Provides official, security-certified Docker containers.

Maintenance & Community

Licensing & Compatibility

  • License: Apache 2.0 License.
  • Compatibility: Permissive license suitable for commercial use and integration into closed-source projects.

Limitations & Caveats

The provided README does not explicitly detail limitations, alpha status, or known bugs.

Health Check
Last Commit

3 days ago

Responsiveness

Inactive

Pull Requests (30d)
17
Issues (30d)
8
Star History
200 stars in the last 30 days

Explore Similar Projects

Feedback? Help us improve.