Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/aws-samples/aws-sagemaker-heart-disease-prediction
This repository contains sample code that is used to demonstrate building, deploying and invoking a SageMaker model for heart disease prediction.
https://github.com/aws-samples/aws-sagemaker-heart-disease-prediction
Last synced: about 2 months ago
JSON representation
This repository contains sample code that is used to demonstrate building, deploying and invoking a SageMaker model for heart disease prediction.
- Host: GitHub
- URL: https://github.com/aws-samples/aws-sagemaker-heart-disease-prediction
- Owner: aws-samples
- License: mit-0
- Created: 2019-11-05T00:45:53.000Z (almost 5 years ago)
- Default Branch: master
- Last Pushed: 2020-10-14T15:34:11.000Z (almost 4 years ago)
- Last Synced: 2024-06-11T01:21:11.020Z (3 months ago)
- Language: Jupyter Notebook
- Homepage:
- Size: 25.3 MB
- Stars: 9
- Watchers: 20
- Forks: 9
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- Contributing: CONTRIBUTING.md
- License: LICENSE
- Code of conduct: CODE_OF_CONDUCT.md
Awesome Lists containing this project
- awesome-aws-research - Amazon SageMaker Heart Disease Prediction
README
## AWS SageMaker Heart Disease Prediction
Machine learning can potentially play a significant role in helping doctors and scientists predict heart disease. A person’s chance of having a heart disease includes many factors such as diabetes, high blood pressure, high cholesterol, abnormal heart rate, and age. In this sample, you will use AWS SageMaker to train a linear learner algorithm that can potentially predict the presence of heart disease.
Note: This is only a sample application and should not be considered as medical advice.
## Architecture
The architecture for this workshop is the following:
## Description
This project uses [Cleveland Heart Disease dataset](https://archive.ics.uci.edu/ml/machine-learning-databases/heart-disease/cleveland.data) taken from the [UCI repository](https://archive.ics.uci.edu/ml/datasets/Heart+Disease). The dataset consists of 303 records that measure an individual's features (age, heart rate, sex, blood pressure, etc.). As depicted above, you will first use SageMaker’s linear learner algorithm to train and deploy a model. Once the model is deployed, you can use AWS API Gateway to get real time prediction for a set of data. If the model predicts the presence of heart disease, an SNS notification is sent to the user e-mail address.
## Quick Start
The quick start guide is intended to deploy the sample application in your own AWS account using a cloud formation template.
Quick Start Setup
Prerequisites:
1. Sign-in to AWS or [Create an Account](https://us-west-2.console.aws.amazon.com)
2. [Create an AWS Bucket](https://docs.aws.amazon.com/AmazonS3/latest/gsg/CreatingABucket.html)
a. Note: please make sure your bucket name starts with ‘sagemaker’. This allows SageMaker to access your bucket.
b. Make a note of the region. Make sure all services used are in the same region as your bucket.
3. Upload **‘heart.csv’** file located in project **/src/main/resources** directory to your AWS Bucket.
4. Upload packaged code **‘heart_function-1.0.0’** provided in root directory to your AWS Bucket.
5. Download **heart-disease-prediction.ipynb** file to your local computer
### Training SageMaker Model
In this section, we will use AWS SageMaker to import the sample jupyter notebook and train a model that can predict heart disease. Once the model is trained, it will be hosted directly on SageMaker.
1. Using AWS Console, select **Amazon SageMaker** from the list of AWS Services.
2. In SageMaker console, select **Notebook instances** from the left navigation panel
3. Select create **notebook instance**
4. Enter a **name** for your notebook. Keep all other fields as default.
5. Click on **Create notebook instance**
6. Once notebook instance is ‘**In Service’**, click on **Open Jupyter.**
7. Click **Upload**
8. Choose **heart-disease-prediction.ipynb**
9. Click on **Upload** again to confirm
10. Click on **heart-disease-prediction.ipynb** to open the file
11. Scroll down to the first section that contains code and enter the **bucket name** that you had created earlier.
12. Once the bucket name is replaced, Click **Run**
13. When a cell blocks finishes execution, a number is displayed next to the cell. **Continue clicking on Run** on each cell block that contains code.
Note: Training and deploying the model take the longest time.
14. After you have hosted your model, copy the **endpoint name.** You will use this endpoint name when launching cloud formation.
15. You can continue to execute the remaining cell blocks to get a prediction on SageMaker notebook.
Warning: Do not execute the ‘(Optional) Delete the endpoint’.
### Launch cloud Formation
In this section, you will launch a cloud formation template that performs the following:
- Creates API Gateway
- Creates a lambda function that calls SageMaker endpoint
- Creates an SNS notification that sends e-mail
1. Using AWS Console, select **CloudFormation** from the list of AWS Services.
2. Choose **Create Stack**.
3. Select **Template is ready** and **Upload a template file**
4. Choose **cloud_formation_template.yaml** file located in project root directory.
5. On the next page, specify stack details
a. Choose a **stack name**
b. Specify your **bucket name** (this is the bucket you created earlier)
c. Specify **Email** address. This email address is used to send you notification if heart disease is predicted.
d. Specify the uploaded **lambda code** (this is the code you uploaded)
e. Specify the S**ageMaker endpoint** (this is the endpoint you copied earlier)
6. Click **Next**
7. On subsequent pages, leave all other fields to their **default** values and click **Next.**
8. On the final page, acknowledge all **Transform might require access capabilities**
9. Choose **Create Stack**
10. Once the Stack is successfully created, click on **Output** tab
11. Copy the **prodDataEndpoint** value. This is the API Gateway endpoint used for real time predictions.
### Predicting Heart disease in real time
The cloud formation template has created an endpoint that can be invoked in real time to predict heart disease. We will use the sample data below to get a prediction of heart disease.
Example data:
You can use API Gateway console or postman, to invoke your API in real time. Below sample data to get heart disease prediction in real time.
{"age": "23","sex": 1,"cp": 3,"trestbps": 145,"chol": 233,"fbs": 1,"restecg": 0,"thalach":150,"exang": 0,"oldpeak": 2.3,"slope": 0,"ca": 0,"thal": 1}
The above call should return a real time prediction:
Note: When prediction label is 1, a notification e-mail is also sent. You may have to first confirm the subscription of your e-mail.
## License
This library is licensed under the MIT-0 License. See the LICENSE file.