Skip to main content

A Machine Learning Brain Stroke Classfier.

Project description

BrainStrokeClassifier API

forthebadge made-with-python
Python 3.9

Functionality of the BrainStrokeClassifier

The package includes a trained Machine Learning model to predict Brain Stroke vulnerability. The model can be accessed as BStClassifier utility. You can use BStClassifier.predict(X) to make predictions. Where X is the input consisting of 7 characteristics:

  1. Gender ("Male", "Female", "Other")
  2. Age
  3. Do you have hypertension (Yes/NO)
  4. Do you have any heart disease (Yes/NO)
  5. What type of work you do? ("Private job", "Self Employed", "Govt. Job", "Never Worked", "You are a Child")
  6. Your average blood glucose level
  7. Your Body Mass Index (BMI)

It classifies whether you are vulnerable to or had brain stroke, or not.

Input type

The input can be:

  1. NumPy array a. 1D array of shape (7,) b. 2D array of shape (S,7); where S is number of samples
  2. Pandas Series of shape (7,)
  3. Pandas DataFrame of shape (S,7); where S is number of columns

About the Machine Learning model

The Machine Learning model is a Pipeline of Strandard Scaler and XGBoost Classifier. You can check the notebook where the model is trained through the github

Package Installation

Make sure you have Python installed in your system. Run Following command in the command window.

 !pip install BrainStrokeClassifier

Example Code

# test.py

# Package installation
!pip install BrainStrokeClassifier

# Loading package
from BrainStrokeClassifier import BStClassifier
import numpy as np
import pandas as pd

# Creating some test input

x = np.array([[0,67,0,1,0,228.69,36.6],
             [0,58,1,0,0,87.96,39.2]])

df = pd.DataFrame(x)
prediction = BStClassifier(df)
print(prediction)

For Suggestions

If you have any suggestions or improvements, please fork the package on github

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

BrainStrokeClassifier-0.1.5.tar.gz (223.9 kB view details)

Uploaded Source

Built Distribution

BrainStrokeClassifier-0.1.5-py3-none-any.whl (345.6 kB view details)

Uploaded Python 3

File details

Details for the file BrainStrokeClassifier-0.1.5.tar.gz.

File metadata

  • Download URL: BrainStrokeClassifier-0.1.5.tar.gz
  • Upload date:
  • Size: 223.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.9.13

File hashes

Hashes for BrainStrokeClassifier-0.1.5.tar.gz
Algorithm Hash digest
SHA256 1e9f84ee93346c35e51aec63a9d8dde42d2f7964cb358ed251f8ac82ad05c9fa
MD5 1d7edd9ff2509858ba95f19c82b733ee
BLAKE2b-256 ca3330bbd12d92dcace1da8f77cdd34166b712ad104fdc79e8b8097ef59299fe

See more details on using hashes here.

File details

Details for the file BrainStrokeClassifier-0.1.5-py3-none-any.whl.

File metadata

File hashes

Hashes for BrainStrokeClassifier-0.1.5-py3-none-any.whl
Algorithm Hash digest
SHA256 6b7a69657f0965eae8646f099367d96cc5dcc2b51836faa2089a51f7921b38e3
MD5 f1279605f71f9732e861da4f1054b468
BLAKE2b-256 23388cb43f6bb8ce5e0058708710edfce9d97b457fc25704b80b469f7ae8bd5a

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page