Skip to content

Latest commit

 

History

History
17 lines (10 loc) · 1.28 KB

README.md

File metadata and controls

17 lines (10 loc) · 1.28 KB

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.