mvpleung / autogluon

AutoGluon: AutoML Toolkit for Deep Learning

AutoGluon: AutoML Toolkit for Deep Learning

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

License

This library is licensed under the Apache 2.0 License.