autogluon简单使用
from autogluon.tabular import TabularDataset, TabularPredictor
train_data = TabularDataset('train.csv')
id, label = 'id', 'label'
# 训练
predictor = TabularPredictor(label=label).fit(
train_data.drop(columns=[id]),
ag_args_fit={'num_gpus': 1},)
# 加载模型
#predictor = TabularPredictor.load("AutogluonModels/ag-20220705_085717/")
# 测试
test_data = TabularDataset('test.csv')
preds = predictor.predict(test_data.drop(columns=[id]))
submission = pd.DataFrame({id:test_data[id], label:preds})
submission.to_csv('submisssion.csv', index=False)