랭킹으로 돌아가기

autogluon/autogluon

Pythonauto.gluon.ai

Fast and Accurate ML in 3 Lines of Code

automlmachine-learningdata-sciencedeep-learningensemble-learningcomputer-visionnatural-language-processingstructured-dataobject-detectiongluontransfer-learningpytorch
스타 성장
스타
10.5k
포크
1.2k
주간 성장
이슈
377
5k10k
2019년 12월2022년 2월2024년 5월2026년 7월
아티팩트PyPIpip install autogluon
README

Fast and Accurate ML in 3 Lines of Code

Latest Release Conda Forge Python Versions Downloads GitHub license Discord Twitter Continuous Integration Platform Tests

Installation | Documentation | Release Notes

AutoGluon, developed by AWS AI, 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.10 - 3.13 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", presets="best")
predictions = predictor.predict("test.csv")
AutoGluon Task Quickstart API
TabularPredictor Quick Start API
TimeSeriesPredictor Quick Start API
MultiModalPredictor Quick Start API

: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: Structured Foundation Models Meets AutoML Expo Talk ICML 2025 2025/07/13
:tv: AutoGluon 1.2: Advancing AutoML with Foundational Models and LLM Agents Expo Workshop NeurIPS 2024 2024/12/10
: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

Articles

Train/Deploy AutoGluon in the Cloud

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

관련 저장소
lukasmasuch/best-of-ml-python

🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

Creative Commons Attribution Share Alike 4.0 Internationalpythonmachine-learning
ml-python.best-of.org
23.7k3.1k
ml-tooling/best-of-ml-python

🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

Creative Commons Attribution Share Alike 4.0 Internationalpythonmachine-learning
ml-python.best-of.org
19.8k2.7k
microsoft/nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

PythonPyPIMIT Licenseautomldeep-learning
nni.readthedocs.io
14.4k1.9k
EpistasisLab/tpot

A Python Automated Machine Learning tool that optimizes machine learning pipelines using genetic programming.

Jupyter NotebookGNU Lesser General Public License v3.0machine-learningpython
epistasislab.github.io/tpot/
10k1.6k
pycaret/pycaret

Open-source, low-code AutoML platform for Python. PyCaret 4.0: sklearn-native engine + React control plane.

PythonPyPIOtherdata-sciencepython
pycaret.org
9.8k1.9k
keras-team/autokeras

AutoML library for deep learning

PythonPyPIApache License 2.0automlneural-architecture-search
autokeras.com
9.3k1.4k
automl/auto-sklearn

Automated Machine Learning with scikit-learn

PythonPyPIBSD 3-Clause "New" or "Revised" Licenseautomlscikit-learn
automl.github.io/auto-sklearn
8.1k1.3k
alteryx/featuretools

An open source python library for automated feature engineering

PythonPyPIBSD 3-Clause "New" or "Revised" Licensefeature-engineeringmachine-learning
featuretools.com
7.7k917
h2oai/h2o-3

H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

Jupyter NotebookApache License 2.0h2omachine-learning
h2o.ai
7.5k2k
google/automl

Google Brain AutoML

Jupyter NotebookApache License 2.0automlefficientdet
6.5k1.5k
zenml-io/zenml

ZenML 🙏: One AI Platform from Pipelines to Agents. https://zenml.io.

PythonPyPIApache License 2.0mlopsmachine-learning
zenml.io
5.5k638
tencentmusic/cube-studio

cube studio开源云原生一站式机器学习/深度学习/大模型AI平台,mlops算法链路全流程,算力租赁平台,notebook在线开发,拖拉拽任务流pipeline编排,多机多卡分布式训练,超参搜索,推理服务VGPU虚拟化,边缘计算,标注平台自动化标注,deepseek等大模型sft微调/奖励模型/强化学习训练,vllm/ollama/mindie大模型多机推理,私有知识库,AI模型市场,支持国产cpu/gpu/npu 昇腾生态,支持RDMA,支持pytorch/tf/mxnet/deepspeed/paddle/colossalai/horovod/ray/volcano等分布式

Otherkubernetesinference
5.1k884