This repository contains code to train and deploy a tensorflow model using aws sagemaker.

data_processing.ipynb file is used to process and store training and test data in an s3 bucket.

train.py file is used to write code used for training the model.

inference_script.py file is used to write script that instructs inference container on where to fetch the model for inference.

train_and_deploy_job.ipynb file is used to orchestrate the training and deployment of the model.

model_artifact_deployment.ipynb file deploys model stored in s3 bucket using aws sageamker.

extend_sagemaker_containers folder contains relevant code for extending sagemaker training containers.