Orawan commited on
Commit
dcb0c79
·
1 Parent(s): d34cbc4

Upload 4 files

Browse files
Files changed (4) hide show
  1. app.ipynb +90 -0
  2. model (1).joblib +3 -0
  3. requirements.txt +5 -0
  4. unique_values (1).joblib +3 -0
app.ipynb ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "nbformat": 4,
3
+ "nbformat_minor": 0,
4
+ "metadata": {
5
+ "colab": {
6
+ "provenance": []
7
+ },
8
+ "kernelspec": {
9
+ "name": "python3",
10
+ "display_name": "Python 3"
11
+ },
12
+ "language_info": {
13
+ "name": "python"
14
+ }
15
+ },
16
+ "cells": [
17
+ {
18
+ "cell_type": "code",
19
+ "execution_count": 1,
20
+ "metadata": {
21
+ "colab": {
22
+ "base_uri": "https://localhost:8080/",
23
+ "height": 140
24
+ },
25
+ "id": "tF2nUGnWA9wQ",
26
+ "outputId": "7b7d0778-02fc-475d-e796-d9a52d773bcc"
27
+ },
28
+ "outputs": [
29
+ {
30
+ "output_type": "error",
31
+ "ename": "IndentationError",
32
+ "evalue": "ignored",
33
+ "traceback": [
34
+ "\u001b[0;36m File \u001b[0;32m\"<ipython-input-1-bfb3c3b35877>\"\u001b[0;36m, line \u001b[0;32m21\u001b[0m\n\u001b[0;31m with st.form(\"questionaire\"):\u001b[0m\n\u001b[0m ^\u001b[0m\n\u001b[0;31mIndentationError\u001b[0m\u001b[0;31m:\u001b[0m unexpected indent\n"
35
+ ]
36
+ }
37
+ ],
38
+ "source": [
39
+ "import joblib\n",
40
+ "import pandas as pd\n",
41
+ "import streamlit as st\n",
42
+ "smoking_status = {'formerly smoked': 1,\n",
43
+ " 'never smoked\t': 2,\n",
44
+ " 'smokes': 3,\n",
45
+ " 'Unknown': 4,\n",
46
+ " }\n",
47
+ "\n",
48
+ "model = joblib.load('model.joblib')\n",
49
+ "unique_values = joblib.load('unique_values.joblib')\n",
50
+ "unique_gender = unique_values[\"gender\"]\n",
51
+ "unique_ever_married = unique_values[\"ever_married\"]\n",
52
+ "unique_work_type = unique_values[\"work_type\"]\n",
53
+ "unique_Residence_type = unique_values[\"Residence_type\"]\n",
54
+ "unique_smoking_status = unique_values[\"smoking_status\"]\n",
55
+ "\n",
56
+ "\n",
57
+ "def main():\n",
58
+ " st.title(\"Adult Income Analysis\")\n",
59
+ " with st.form(\"questionaire\"):\n",
60
+ " age = st.slider(\"age\", min_value=0, max_value=100)\n",
61
+ " gender = st.selectbox(\"gender\", unique_gender)\n",
62
+ " hypertension = st.slider(\"hypertension\", min_value=0, max_value=1)\n",
63
+ " heart_disease = st.slider(\"heart_disease\", min_value=0, max_value=1)\n",
64
+ " ever_married = st.selectbox(\"ever_married\", unique_ever_married)\n",
65
+ " work_type = st.selectbox(\"work_type\", unique_work_type)\n",
66
+ " Residence_type = st.selectbox(\"Residence_type\", unique_Residence_type)\n",
67
+ " avg_glucose_level = st.slider(\"avg_glucose_level\", min_value=0, max_value=300)\n",
68
+ " bmi = st.slider(\"bmi\", min_value=0, max_value=100)\n",
69
+ " smoking_status = st.selectbox(\"smoking_status\", unique_smoking_status)\n",
70
+ "\n",
71
+ "clicked = st.form_submit_button(\"Predict stroke\")\n",
72
+ "if clicked:\n",
73
+ " result=model.predict(pd.DataFrame({\"age\": [age],\n",
74
+ " \"gender\": [gender],\n",
75
+ " \"hypertension\": [hypertension],\n",
76
+ " \"heart_disease\": [heart_disease],\n",
77
+ " \"ever_married\": [ever_married],\n",
78
+ " \"work_type\": [work_type],\n",
79
+ " \"Residence_type\": [Residence_type],\n",
80
+ " \"avg_glucose_level\": [avg_glucose_level],\n",
81
+ " \"bmi\": [bmi],\n",
82
+ " \"smoking_status\":[smoking_status]}))\n",
83
+ " result = '1' if result[0] == 1 else '0'\n",
84
+ " st.success('The predicted stroke is {}'.format(result))\n",
85
+ "if __name__=='__main__':\n",
86
+ " main()"
87
+ ]
88
+ }
89
+ ]
90
+ }
model (1).joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6074ed90c8d54cda47c2f4649c85593f868ca8f25e25fcd3cf3f8b9e1af01ea4
3
+ size 76513
requirements.txt ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ joblib
2
+ pandas
3
+ scikit-learn==1.2.2
4
+ xgboost==1.7.6
5
+ altair<5
unique_values (1).joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d50a7d378062a047249d0a3a123de19866e62dcf3d47aae7acbded938be46d45
3
+ size 1425