Skip to main content

AutoML Toolkit with MXNet Gluon

Project description

AutoML Toolkit for Deep Learning

Build Status Pypi Version Upload Python Package

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 tabular, image, and text data.

Example

# First install package from terminal:
# python3 -m pip install --upgrade pip
# python3 -m pip install --upgrade setuptools
# python3 -m pip install --upgrade "mxnet<2.0.0"
# python3 -m pip install autogluon

from autogluon.tabular 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)

Resources

See the AutoGluon Website for documentation and instructions on:

Scientific Publications

Articles

Hands-on Tutorials

Train/Deploy AutoGluon in the Cloud

Citing AutoGluon

If you use AutoGluon in a scientific publication, please cite the following paper:

Erickson, Nick, et al. "AutoGluon-Tabular: Robust and Accurate AutoML for Structured Data." arXiv preprint arXiv:2003.06505 (2020).

BibTeX entry:

@article{agtabular,
  title={AutoGluon-Tabular: Robust and Accurate AutoML for Structured Data},
  author={Erickson, Nick and Mueller, Jonas and Shirkov, Alexander and Zhang, Hang and Larroy, Pedro and Li, Mu and Smola, Alexander},
  journal={arXiv preprint arXiv:2003.06505},
  year={2020}
}

AutoGluon for Hyperparameter and Neural Architecture Search (HNAS)

AutoGluon also provides state-of-the-art tools for neural hyperparameter and architecture search, such as for example ASHA, Hyperband, Bayesian Optimization and BOHB. To get started, checkout the following resources

Also have a look at our paper "Model-based Asynchronous Hyperparameter and Neural Architecture Search" arXiv preprint arXiv:2003.10865 (2020).

@article{abohb,
  title={Model-based Asynchronous Hyperparameter and Neural Architecture Search},
  author={Klein, Aaron and Tiao, Louis and Lienart, Thibaut and Archambeau, Cedric and Seeger, Matthias},
  journal={arXiv preprint arXiv:2003.10865},
  year={2020}
}

License

This library is licensed under the Apache 2.0 License.

Contributing to AutoGluon

We are actively accepting code contributions to the AutoGluon project. If you are interested in contributing to AutoGluon, please read the Contributing Guide to get started.

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

autogluon.text-0.0.15b20201015.tar.gz (52.1 kB view details)

Uploaded Source

Built Distribution

autogluon.text-0.0.15b20201015-py3-none-any.whl (59.4 kB view details)

Uploaded Python 3

File details

Details for the file autogluon.text-0.0.15b20201015.tar.gz.

File metadata

  • Download URL: autogluon.text-0.0.15b20201015.tar.gz
  • Upload date:
  • Size: 52.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.1.0 requests-toolbelt/0.9.1 tqdm/4.50.2 CPython/3.7.9

File hashes

Hashes for autogluon.text-0.0.15b20201015.tar.gz
Algorithm Hash digest
SHA256 56cdb96828d621e94bc1afccec8a99b5c1469251c262fd74c8d52a3f646d6e92
MD5 c94aa28e117357142ec048de2d846895
BLAKE2b-256 16bfed07294bfa35c3682ce6823abdec6e40767f2b79806cf9184fd9a32845b5

See more details on using hashes here.

File details

Details for the file autogluon.text-0.0.15b20201015-py3-none-any.whl.

File metadata

  • Download URL: autogluon.text-0.0.15b20201015-py3-none-any.whl
  • Upload date:
  • Size: 59.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.1.0 requests-toolbelt/0.9.1 tqdm/4.50.2 CPython/3.7.9

File hashes

Hashes for autogluon.text-0.0.15b20201015-py3-none-any.whl
Algorithm Hash digest
SHA256 a1ec7ced7e9fb3a1e7e7a3d28a338e9471ada05dfdb2f2effea2460cd53096af
MD5 9e9ee6d06ca1a8be26b441b11c8f290e
BLAKE2b-256 8b03034a784b3c5f5b96cd7749829c3ceb618991d09de913c66c7838ef43c21c

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page