AutoGluon使用帮助

最近在看李沐大神的动手学深度学习的视频,看到大神努力地推荐AutoGluon,说它能够让普通人很容易上手使用深度学习模型,因此学习一下。

介绍

AutoGluon由AWS AI开发,自动化机器学习任务,使您能够轻松地在应用程序中实现强大的预测性能。只需几行代码,您就可以在图像、文本、时间序列和表格数据上训练和部署高精度的机器学习和深度学习模型。

安装

使用conda创建一个环境,再在该环境上安装AutoGluon。

1
2
3
4
5
conda create -n autogluon python=3.11
conda activate autogluon
conda install -c conda-forge mamba
mamba install -c conda-forge -c pytorch -c nvidia autogluon "pytorch=*=*cuda*"
mamba install -c conda-forge "ray-tune >=2.10.0,<2.32" "ray-default >=2.10.0,<2.32" # install ray for faster training

测试

使用如下代码快速演示:

1
2
3
from autogluon.tabular import TabularPredictor
predictor = TabularPredictor(label="class").fit("train.csv")
predictions = predictor.predict("test.csv")

参考链接

  1. 使用开源 AutoML 库 AutoGluon 进行机器学习,by Shashank Prasanna.
  2. AutoGluon,一个强大的 Python 库,by 算法金.
  3. AutoGluon-教程1-简单的入门模型,by 愿航.
  4. autogluon,by github.
  5. Installing AutoGluon,by autogluon.