autogluon_tabular_classification_quickstart_train_predict_evaluate.py

1from autogluon.tabular import TabularDataset, TabularPredictor
2
3# Load training data from a public S3 bucket
4train_data = TabularDataset('https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv')
5
6# Initialize and train the predictor
7# 'class' is the name of the column we want to predict
8label = 'class'
9predictor = TabularPredictor(label=label).fit(train_data)
10
11# Load test data to evaluate the model
12test_data = TabularDataset('https://autogluon.s3.amazonaws.com/datasets/Inc/test.csv')
13
14# Make predictions and view the results
15y_pred = predictor.predict(test_data)
16print(y_pred.head())
17
18# Evaluate the model performance
19perf = predictor.evaluate(test_data)
20print(perf)