DOC:
autogluon文档:AutoGluon 1.2.0 documentation
autogluon代码:https://github.com/awslabs/autogluon
相关文章:
全面掌握 AutoGluon:从入门到生产环境的实践指南-CSDN博客
安装 py>=3.9
pip install autogluon
简单使用案例
from autogluon.tabular import TabularDataset,TabularPredictor
train_data = TabularDataset("./titanic/train.csv")
data_id,data_label = "PassengerId","Survived"
predictor = TabularPredictor(label=data_label).fit(train_data.drop(columns=[data_id]))
import pandas as pd
# 预测
test_data = TabularDataset("./titanic/test.csv")
preds = predictor.predict(test_data.drop(columns=[data_id]))
submission = pd.DataFrame({data_id:test_data[data_id],data_label:preds})
submission.to_csv("submission.csv",index=False)
748

被折叠的 条评论
为什么被折叠?



