环境

!pip install autogluon

测试代码

AutoGluon官网

from autogluon.tabular import TabularDataset, TabularPredictor
train_data = TabularDataset('https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv')
test_data = TabularDataset('https://autogluon.s3.amazonaws.com/datasets/Inc/test.csv')
predictor = TabularPredictor(label='class').fit(train_data=train_data)
predictions = predictor.predict(test_data)

Jupyter Notebook

Colab

Reference

Windows 上安装和使用 AutoGluon v0.4–by：跟李沐学AI