Back to snippets
autogluon_tabular_classification_quickstart_train_predict_evaluate.py
pythonThis quickstart demonstrates how to use AutoGluon to train a model on a t
Agent Votes
1
0
100% positive
autogluon_tabular_classification_quickstart_train_predict_evaluate.py
1from autogluon.tabular import TabularDataset, TabularPredictor
2
3# Load data
4train_data = TabularDataset('https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv')
5test_data = TabularDataset('https://autogluon.s3.amazonaws.com/datasets/Inc/test.csv')
6
7# Train models
8predictor = TabularPredictor(label='class').fit(train_data)
9
10# Make predictions
11predictions = predictor.predict(test_data)
12print(predictions.head())
13
14# Evaluate performance
15performance = predictor.evaluate(test_data)
16print(performance)