AutoGluon automates machine learning tasks enabling you to easily achieve strong predictive performance in your applications. With just a few lines of code, you can train and deploy high-accuracy deep learning models on image, text, and tabular data. Get started with:
# First install package from terminal: pip install mxnet autogluon
from autogluon import TabularPrediction as task
train_data = task.Dataset(file_path='https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv')
test_data = task.Dataset(file_path='https://autogluon.s3.amazonaws.com/datasets/Inc/test.csv')
predictor = task.fit(train_data=train_data, label='class')
performance = predictor.evaluate(test_data)
See the AutoGluon Website for instructions on:
- Installing AutoGluon
- Learning with image data
- Learning with text data
- Learning with data in tabular format
- More advanced topics such as Neural Architecture Search
This library is licensed under the Apache 2.0 License.