AutoML with AutoGluon, Amazon SageMaker, and AWS Lambda

This repository contains the CloudFormation template and prewritten source code powering the code-free AutoML pipeline detailed in this AWS Machine Learning blog post. Feel free to customize it to fit your use case and share with us what you build!

• autogluon-tab-with-test.py is the script run by the SageMaker training job the Lambda function automatically kicks off when you upload your training data to S3. It's pre-packaged in sourcedir.tar.gz for the use of the pipeline. You can modify this script to reuse the pipeline with your own model training code.
• CodeFreeAutoML.yaml is the CloudFormation template you use to deploy the pipeline in your account.
• lambda_function.py is the source code for the Lambda function that kicks off the SageMaker training job when you upload your data to S3.
• sourcedir.tar.gz is the autogluon-tab-with-test.py file pre-packaged for your convenience; the pipeline requires it to be gzipped.

Security

See CONTRIBUTING for more information.

License

This project is licensed under the Apache-2.0 License.