Skip to content
AutoGluon: AutoML Toolkit for Deep Learning
Branch: master
Clone or download
jwmueller Improve NN runtime by reducing num_categories_to_keep (#246)
* changed default num_categories_to_keep hyperparameter of NN to improve runtimes

* allow NN to be trained for 0 epochs
Latest commit 93a00ed Jan 28, 2020

AutoGluon: AutoML Toolkit for Deep Learning

Build Status Pypi Version

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='')
test_data = task.Dataset(file_path='')
predictor =, label='class')
performance = predictor.evaluate(test_data)

See the AutoGluon Website for instructions on:


This library is licensed under the Apache 2.0 License.

You can’t perform that action at this time.