autogluon

三行代码搞定高精度模型,自动调参集成,省时省力

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 模型训练 🍴 复刻 1218 🐛 待处理 458 👀 关注 99
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 10822
维护状态 活跃
是否开源 开源
定价模式 🏷️完全免费 开源项目,Apache-2.0许可,完全免费,可自行部署使用。

项目预览

autogluon 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类模型训练
开发团队autogluon
定价模式🏷️完全免费
价格说明开源项目,Apache-2.0许可,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型autogluon,automated-machine-learning,automl,computer-vision,data-science,deep-learning,ensemble-learning,forecasting,gluon,hyperparameter-optimization,machine-learning,natural-language-processing,object-detection,python,pytorch,scikit-learn,structured-data,tabular-data,time-series,transfer-learning
GitHub 星标★ 10822
上线时间2019-07-29 00:00:00
代码更新2026-11-09
仓库体积27.4 MB
本站数据更新2026-11-10
维护状态活跃
收录时间2026-08-10
浏览次数7

项目介绍

AutoGluon 是一个自动化机器学习(AutoML)库,旨在用三行代码实现快速且准确的模型训练。它解决了传统机器学习中模型选择、超参数调优和特征工程耗时费力的问题,通过自动化流程,让非专家也能轻松构建高性能模型。核心能力包括自动数据预处理、模型集成(如 stacking 和 bagging)、超参数优化,以及支持表格数据、计算机视觉、自然语言处理和时间序列预测等多种任务。用户只需提供数据,AutoGluon 会自动训练多个模型并集成,通常能取得优于手动调参的效果。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-10 · 当前星标 ★ 10822

使用教程

难度:入门 约 10 分钟 部署方式:库/依赖 4 步

环境要求

  • Python 3.10 - 3.13
  • 操作系统:Linux、MacOS 或 Windows
  • 可用的 pip 包管理环境
  • 本地已有或自行准备的训练/测试数据文件(如 train.csv、test.csv),且训练文件中包含目标列

安装与启动步骤

  1. 1检查 Python 版本

    先确认本机 Python 版本处于 README 声明的 3.10 - 3.13 区间,版本不符需先切换环境。

    python --version
  2. 2安装 AutoGluon

    用 pip 一行安装完整版 AutoGluon。README 说明 GPU 支持、Conda 安装与可选依赖需查阅官方安装指南。

    pip install autogluon
  3. 3准备数据文件

    准备 train.csv 与 test.csv,放在运行脚本的同目录;train.csv 中必须含目标列(示例中列名为 class)。

  4. 4运行三行代码训练

    按 README Quickstart 写入脚本并执行:拟合训练集、再对测试集做预测。label 换成你数据中真实的标签列名。

    python quickstart.py

关键配置

配置项必填说明示例
label是目标列名,必须与 train.csv 中实际列名一致class
presets否拟合预设参数,示例使用 bestbest
train.csv是fit 使用的训练数据路径train.csv
test.csv是predict 使用的待预测数据路径test.csv

如何确认成功

脚本无报错执行完毕,predictor.predict("test.csv") 正常返回对测试集的预测结果即可。

常见问题

Q:安装时提示 Python 版本不支持怎么办?

A:README 明确 AutoGluon 仅支持 Python 3.10 - 3.13,请切换到该区间内的解释器后再执行 pip install autogluon。

Q:如何在 GPU 上安装,或改用 Conda 安装?

A:README 未给出具体命令,只提供了官方 Installation Guide 链接,GPU 支持、Conda 安装与可选依赖请按该指南操作。

Q:如何把 AutoGluon 部署到云端或容器?

A:README 列出 AutoGluon Cloud、深度学习容器、官方 Docker 镜像与 SageMaker Autopilot 入口,但未给出具体命令,需按对应链接指引操作。

Q:运行时提示找不到 label 列?

A:Quickstart 中的 class 只是示例,需把 label 改成 train.csv 里真实存在的目标列名。

注意事项

  • README 中的三行代码针对表格任务,时间序列和多模态任务请参考对应 Quickstart 链接。
  • README 未给出 Docker 运行命令、端口或镜像标签,如需容器部署请查阅官方文档后再操作。
  • 示例中的 train.csv、test.csv 为相对路径,运行脚本时需保证工作目录与数据文件一致。

核心亮点

  • 自动进行数据清洗、特征工程和模型集成,新手也能获得顶尖精度
  • 支持表格、图像、文本和时间序列多模态任务,覆盖广泛
  • 训练过程高度自动化,无需手动调参,显著降低使用门槛

不足之处

  • 自动搜索和集成过程计算开销大,训练速度较慢
  • 高度自动化导致对底层模型的定制灵活性不足

适用场景

  • 数据科学竞赛中快速构建高精度基线模型
  • 企业业务预测和分类任务,缺乏专业ML团队时使用
  • 多模态数据(图像+文本)的自动化建模场景

替代项目

H2O AutoML、TPOT、AutoKeras

上一篇:hud-python

下一篇:catalyst

同类项目推荐

benderopt 开源

无梯度也能调参,黑盒优化一键搞定

Black-box optimization library

★ 87 2026-08-09
Dreambooth-Stable-Diffusion 开源

用几张照片,训练专属 Stable Diffusion 模型,生成你的定制图像。

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) by way of Textual In···

★ 3214 2026-08-09
MoE-plus-plus 开源

零计算专家,让MoE模型更快更省,性能不减

[ICLR 2025] MoE++: Accelerating Mixture-of-Experts Methods with Zero-Computation Exp···

★ 281 2026-08-09
LRV-Instruction 开源

减少多模态模型胡说八道,让AI描述更靠谱

[ICLR'24] Mitigating Hallucination in Large Multi-Modal Models via Robust Instructio···

★ 297 2026-08-09
JacobiForcing 开源

让模型生成快数倍,质量不打折的并行解码新方案

[ICML 2026] Jacobi Forcing: Fast and Accurate Diffusion-style Decoding

★ 119 2026-08-09
DPO-Shift 开源

修正 DPO 分布偏移,让模型对齐更稳更准

DPO-Shift: Shifting the Distribution of Direct Preference Optimization

★ 59 2026-08-09
CDM 开源

少步扩散蒸馏,让图像生成快如闪电

[NeurIPS 2026] Continuous-Time Distribution Matching for Few-Step Diffusion Distilla···

★ 156 2026-08-09
FeynRL 开源

换目标、换采样,后训练实验一套框架跑通

Post-training framework for large models, from new objectives to new rollout systems···

★ 228 2026-10-31