Contents
Stroke is a medical condition in which poor blood flow to the brain results in cell death. There are two main types of stroke: ischemic, due to lack of blood flow, and hemorrhagic, due to bleeding. Both result in parts of the brain not functioning properly.Prediction model helps to identify the person suffering from Stroke by taking into account various observational parameters and demographic details.
The csv file used in the model has 10 columns viz
Same kind of CSV file should be fed to the model for best output.
For reference, A sample CSV file is attached in the link. The user only needs to clik on the link to get it downloaded to the local. The data should be in a similar format like the sample csv.
Download sample Data from the link : Data Source for Stroke
import pandas as pd
df = pd.read_csv(' S3/Sagemaker path of Stroke.csv file')
df.to_csv("sample.csv",index=False)
df.head(3)
The model can be directly accessed via the console provided by AWS. However, for a more customized process one can access the model using the below code as well.
To create a model, import boto3, sagemaker and get the arn of the model package
import boto3
import sagemaker
role = sagemaker.get_execution_role()
smmp = boto3.client('sagemakermp')
modelName='Name of the model'
modelArn = 'Model ARN name'
createHeatIndexResponse = smmp.create_model(ModelName=modelName,\
Containers=[{'ModelPackageName': modelArn}],\
ExecutionRoleArn=role,\
EnableNetworkIsolation=True )
configName ='<Input Configuration Name>'
instanceType = '<Input Instance Type>'
createHeatIndexEndpointConfig = smmp.create_endpoint_config(EndpointConfigName = configName, ProductionVariants = [{'InstanceType':instanceType, 'InitialInstanceCount':1, 'ModelName':modelName, 'VariantName':'xyz'}])
endpointName = '<Input Endpoint Name>'
createHeatIndexEndpoint = smmp.create_endpoint(EndpointName = endpointName, EndpointConfigName = configName)
runtime = boto3.Session().client('runtime.sagemaker')
#Reading Input Data
with open('sample.csv','rb') as f:
payload = f.read()
response = runtime.invoke_endpoint(EndpointName = endpointName, ContentType = 'text/csv', Body = payload)
result = response['Body'].read().decode()
#Writing Output Data
with open('sampleOutput.txt','w') as f:
f.write(result)
If you're ready to be done with this notebook, please run the delete_endpoint line in the cell below. This will remove the hosted endpoint you created and avoid any charges from a stray instance being left on
sagemaker.Session().delete_endpoint(endpointName)
with open("./sampleOutput.txt","r") as f:
sampleResponse =f.read()
sampleResponse = sampleResponse.split('\n')
sampleResponse