P4_NeuralNetwork.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Neural Network Model"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 266,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [],
   "source": [
    "import jenkspy as jenkspy\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "import seaborn as sns\n",
    "from sklearn.ensemble import RandomForestClassifier\n",
    "from sklearn.metrics import accuracy_score\n",
    "from sklearn.model_selection import train_test_split\n",
    "import matplotlib.pyplot as plt\n",
    "from sklearn import tree, metrics\n",
    "from sklearn.metrics import confusion_matrix, classification_report\n",
    "from sklearn.neural_network import MLPClassifier"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 267,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>age</th>\n",
       "      <th>hypertension</th>\n",
       "      <th>heart_disease</th>\n",
       "      <th>avg_glucose_level</th>\n",
       "      <th>bmi</th>\n",
       "      <th>work_type_Govt_job</th>\n",
       "      <th>work_type_Never_worked</th>\n",
       "      <th>work_type_Private</th>\n",
       "      <th>work_type_Self-employed</th>\n",
       "      <th>smoking_status_Unknown</th>\n",
       "      <th>...</th>\n",
       "      <th>smoking_status_never smoked</th>\n",
       "      <th>smoking_status_smokes</th>\n",
       "      <th>gender_Female</th>\n",
       "      <th>gender_Male</th>\n",
       "      <th>gender_Other</th>\n",
       "      <th>ever_married_No</th>\n",
       "      <th>ever_married_Yes</th>\n",
       "      <th>Residence_type_Rural</th>\n",
       "      <th>Residence_type_Urban</th>\n",
       "      <th>stroke</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0.931846</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>2.500974</td>\n",
       "      <td>0.859059</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1.673346</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>-0.056516</td>\n",
       "      <td>0.279850</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>-0.094846</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1.303993</td>\n",
       "      <td>0.548264</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>1.616307</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1.364196</td>\n",
       "      <td>-0.920950</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>1.730384</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1.616050</td>\n",
       "      <td>-0.214597</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>1.331115</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>-0.802911</td>\n",
       "      <td>-0.440630</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>1.045923</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.296704</td>\n",
       "      <td>-1.090475</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>1.559269</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-1.042890</td>\n",
       "      <td>-0.892696</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>1.730384</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.587512</td>\n",
       "      <td>-0.115708</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>0.589615</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0.246375</td>\n",
       "      <td>0.887313</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>0.190346</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.085888</td>\n",
       "      <td>-0.454757</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>1.616307</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>2.196834</td>\n",
       "      <td>-0.327614</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>-0.037808</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1.224417</td>\n",
       "      <td>0.053817</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>0.760730</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1.728540</td>\n",
       "      <td>0.986203</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>1.388153</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>2.346821</td>\n",
       "      <td>-0.666663</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>0.532577</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.404403</td>\n",
       "      <td>1.028584</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>1.160000</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1.777078</td>\n",
       "      <td>-1.146983</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>0.076269</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>2.596799</td>\n",
       "      <td>2.596687</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>1.616307</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>2.501183</td>\n",
       "      <td>-0.553646</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>1.787423</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>2.076219</td>\n",
       "      <td>0.279850</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>1.160000</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.120052</td>\n",
       "      <td>-0.468884</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>1.673346</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.094013</td>\n",
       "      <td>-0.991585</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>0.817769</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.159424</td>\n",
       "      <td>-0.327614</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>1.045923</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1.803950</td>\n",
       "      <td>-0.313487</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>0.361461</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>2.154962</td>\n",
       "      <td>1.932715</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>-0.494116</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.525434</td>\n",
       "      <td>-0.723171</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>1.787423</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1.839156</td>\n",
       "      <td>-1.175237</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>1.673346</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>3.001557</td>\n",
       "      <td>-0.002691</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>-0.151885</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.508977</td>\n",
       "      <td>-0.115708</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>1.787423</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>-0.512519</td>\n",
       "      <td>-0.567774</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>30 rows × 21 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "         age  hypertension  heart_disease  avg_glucose_level       bmi  \\\n",
       "0   0.931846             0              1           2.500974  0.859059   \n",
       "1   1.673346             0              1          -0.056516  0.279850   \n",
       "2  -0.094846             0              0           1.303993  0.548264   \n",
       "3   1.616307             1              0           1.364196 -0.920950   \n",
       "4   1.730384             0              0           1.616050 -0.214597   \n",
       "5   1.331115             1              1          -0.802911 -0.440630   \n",
       "6   1.045923             0              0          -0.296704 -1.090475   \n",
       "7   1.559269             0              0          -1.042890 -0.892696   \n",
       "8   1.730384             1              0          -0.587512 -0.115708   \n",
       "9   0.589615             0              1           0.246375  0.887313   \n",
       "10  0.190346             0              0          -0.085888 -0.454757   \n",
       "11  1.616307             0              1           2.196834 -0.327614   \n",
       "12 -0.037808             1              0           1.224417  0.053817   \n",
       "13  0.760730             0              1           1.728540  0.986203   \n",
       "14  1.388153             1              0           2.346821 -0.666663   \n",
       "15  0.532577             0              0          -0.404403  1.028584   \n",
       "16  1.160000             0              0           1.777078 -1.146983   \n",
       "17  0.076269             1              0           2.596799  2.596687   \n",
       "18  1.616307             0              0           2.501183 -0.553646   \n",
       "19  1.787423             0              1           2.076219  0.279850   \n",
       "20  1.160000             0              0          -0.120052 -0.468884   \n",
       "21  1.673346             0              0          -0.094013 -0.991585   \n",
       "22  0.817769             0              0          -0.159424 -0.327614   \n",
       "23  1.045923             0              1           1.803950 -0.313487   \n",
       "24  0.361461             1              0           2.154962  1.932715   \n",
       "25 -0.494116             0              0          -0.525434 -0.723171   \n",
       "26  1.787423             1              0           1.839156 -1.175237   \n",
       "27  1.673346             0              1           3.001557 -0.002691   \n",
       "28 -0.151885             0              0          -0.508977 -0.115708   \n",
       "29  1.787423             1              1          -0.512519 -0.567774   \n",
       "\n",
       "    work_type_Govt_job  work_type_Never_worked  work_type_Private  \\\n",
       "0                    0                       0                  1   \n",
       "1                    0                       0                  1   \n",
       "2                    0                       0                  1   \n",
       "3                    0                       0                  0   \n",
       "4                    0                       0                  1   \n",
       "5                    0                       0                  1   \n",
       "6                    0                       0                  1   \n",
       "7                    0                       0                  1   \n",
       "8                    0                       0                  1   \n",
       "9                    1                       0                  0   \n",
       "10                   0                       0                  1   \n",
       "11                   0                       0                  1   \n",
       "12                   0                       0                  0   \n",
       "13                   0                       0                  1   \n",
       "14                   0                       0                  1   \n",
       "15                   0                       0                  1   \n",
       "16                   1                       0                  0   \n",
       "17                   0                       0                  0   \n",
       "18                   0                       0                  0   \n",
       "19                   0                       0                  1   \n",
       "20                   0                       0                  1   \n",
       "21                   0                       0                  0   \n",
       "22                   0                       0                  1   \n",
       "23                   0                       0                  0   \n",
       "24                   0                       0                  1   \n",
       "25                   0                       0                  1   \n",
       "26                   0                       0                  0   \n",
       "27                   0                       0                  0   \n",
       "28                   1                       0                  0   \n",
       "29                   0                       0                  1   \n",
       "\n",
       "    work_type_Self-employed  smoking_status_Unknown  ...  \\\n",
       "0                         0                       0  ...   \n",
       "1                         0                       0  ...   \n",
       "2                         0                       0  ...   \n",
       "3                         1                       0  ...   \n",
       "4                         0                       0  ...   \n",
       "5                         0                       0  ...   \n",
       "6                         0                       0  ...   \n",
       "7                         0                       1  ...   \n",
       "8                         0                       0  ...   \n",
       "9                         0                       0  ...   \n",
       "10                        0                       0  ...   \n",
       "11                        0                       0  ...   \n",
       "12                        1                       0  ...   \n",
       "13                        0                       0  ...   \n",
       "14                        0                       0  ...   \n",
       "15                        0                       0  ...   \n",
       "16                        0                       0  ...   \n",
       "17                        1                       0  ...   \n",
       "18                        1                       0  ...   \n",
       "19                        0                       1  ...   \n",
       "20                        0                       0  ...   \n",
       "21                        1                       0  ...   \n",
       "22                        0                       0  ...   \n",
       "23                        1                       0  ...   \n",
       "24                        0                       0  ...   \n",
       "25                        0                       1  ...   \n",
       "26                        1                       0  ...   \n",
       "27                        1                       0  ...   \n",
       "28                        0                       0  ...   \n",
       "29                        0                       0  ...   \n",
       "\n",
       "    smoking_status_never smoked  smoking_status_smokes  gender_Female  \\\n",
       "0                             0                      0              0   \n",
       "1                             1                      0              0   \n",
       "2                             0                      1              1   \n",
       "3                             1                      0              1   \n",
       "4                             0                      0              0   \n",
       "5                             1                      0              0   \n",
       "6                             1                      0              1   \n",
       "7                             0                      0              1   \n",
       "8                             1                      0              1   \n",
       "9                             0                      1              1   \n",
       "10                            0                      1              1   \n",
       "11                            1                      0              1   \n",
       "12                            1                      0              1   \n",
       "13                            0                      1              0   \n",
       "14                            0                      1              0   \n",
       "15                            1                      0              1   \n",
       "16                            0                      1              1   \n",
       "17                            1                      0              1   \n",
       "18                            1                      0              1   \n",
       "19                            0                      0              0   \n",
       "20                            0                      0              0   \n",
       "21                            1                      0              0   \n",
       "22                            0                      0              1   \n",
       "23                            0                      1              0   \n",
       "24                            0                      1              0   \n",
       "25                            0                      0              0   \n",
       "26                            1                      0              1   \n",
       "27                            0                      0              0   \n",
       "28                            1                      0              0   \n",
       "29                            0                      0              1   \n",
       "\n",
       "    gender_Male  gender_Other  ever_married_No  ever_married_Yes  \\\n",
       "0             1             0                0                 1   \n",
       "1             1             0                0                 1   \n",
       "2             0             0                0                 1   \n",
       "3             0             0                0                 1   \n",
       "4             1             0                0                 1   \n",
       "5             1             0                0                 1   \n",
       "6             0             0                1                 0   \n",
       "7             0             0                0                 1   \n",
       "8             0             0                0                 1   \n",
       "9             0             0                0                 1   \n",
       "10            0             0                0                 1   \n",
       "11            0             0                0                 1   \n",
       "12            0             0                0                 1   \n",
       "13            1             0                0                 1   \n",
       "14            1             0                0                 1   \n",
       "15            0             0                1                 0   \n",
       "16            0             0                0                 1   \n",
       "17            0             0                0                 1   \n",
       "18            0             0                0                 1   \n",
       "19            1             0                0                 1   \n",
       "20            1             0                0                 1   \n",
       "21            1             0                0                 1   \n",
       "22            0             0                0                 1   \n",
       "23            1             0                0                 1   \n",
       "24            1             0                0                 1   \n",
       "25            1             0                0                 1   \n",
       "26            0             0                0                 1   \n",
       "27            1             0                0                 1   \n",
       "28            1             0                1                 0   \n",
       "29            0             0                1                 0   \n",
       "\n",
       "    Residence_type_Rural  Residence_type_Urban  stroke  \n",
       "0                      0                     1       1  \n",
       "1                      1                     0       1  \n",
       "2                      0                     1       1  \n",
       "3                      1                     0       1  \n",
       "4                      0                     1       1  \n",
       "5                      1                     0       1  \n",
       "6                      0                     1       1  \n",
       "7                      0                     1       1  \n",
       "8                      1                     0       1  \n",
       "9                      1                     0       1  \n",
       "10                     0                     1       1  \n",
       "11                     0                     1       1  \n",
       "12                     1                     0       1  \n",
       "13                     0                     1       1  \n",
       "14                     0                     1       1  \n",
       "15                     0                     1       1  \n",
       "16                     1                     0       1  \n",
       "17                     0                     1       1  \n",
       "18                     0                     1       1  \n",
       "19                     1                     0       1  \n",
       "20                     0                     1       1  \n",
       "21                     1                     0       1  \n",
       "22                     1                     0       1  \n",
       "23                     0                     1       1  \n",
       "24                     0                     1       1  \n",
       "25                     1                     0       1  \n",
       "26                     0                     1       1  \n",
       "27                     0                     1       1  \n",
       "28                     0                     1       1  \n",
       "29                     1                     0       1  \n",
       "\n",
       "[30 rows x 21 columns]"
      ]
     },
     "execution_count": 267,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df=pd.read_csv('prepared_data.csv')\n",
    "df.head(30)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 268,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[-1.8060002037748208, -1.5208079189339547, -1.1785771771249152, -0.8363464353158757, -0.4941156935068361, -0.1518849516977966, 0.1903457901112428, 0.5325765319202823, 0.9318457306974952, 1.3881533864428812, 1.7874225852200938]\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>age</th>\n",
       "      <th>hypertension</th>\n",
       "      <th>heart_disease</th>\n",
       "      <th>avg_glucose_level</th>\n",
       "      <th>bmi</th>\n",
       "      <th>work_type_Govt_job</th>\n",
       "      <th>work_type_Never_worked</th>\n",
       "      <th>work_type_Private</th>\n",
       "      <th>work_type_Self-employed</th>\n",
       "      <th>smoking_status_Unknown</th>\n",
       "      <th>...</th>\n",
       "      <th>smoking_status_never smoked</th>\n",
       "      <th>smoking_status_smokes</th>\n",
       "      <th>gender_Female</th>\n",
       "      <th>gender_Male</th>\n",
       "      <th>gender_Other</th>\n",
       "      <th>ever_married_No</th>\n",
       "      <th>ever_married_Yes</th>\n",
       "      <th>Residence_type_Rural</th>\n",
       "      <th>Residence_type_Urban</th>\n",
       "      <th>stroke</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>8</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>2.500974</td>\n",
       "      <td>0.859059</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>-0.056516</td>\n",
       "      <td>0.279850</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1.303993</td>\n",
       "      <td>0.548264</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>10</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1.364196</td>\n",
       "      <td>-0.920950</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1.616050</td>\n",
       "      <td>-0.214597</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>9</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>-0.802911</td>\n",
       "      <td>-0.440630</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>9</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.296704</td>\n",
       "      <td>-1.090475</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-1.042890</td>\n",
       "      <td>-0.892696</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>10</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.587512</td>\n",
       "      <td>-0.115708</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>8</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0.246375</td>\n",
       "      <td>0.887313</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.085888</td>\n",
       "      <td>-0.454757</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>2.196834</td>\n",
       "      <td>-0.327614</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>6</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1.224417</td>\n",
       "      <td>0.053817</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>8</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1.728540</td>\n",
       "      <td>0.986203</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>9</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>2.346821</td>\n",
       "      <td>-0.666663</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>7</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.404403</td>\n",
       "      <td>1.028584</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>9</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1.777078</td>\n",
       "      <td>-1.146983</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>6</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>2.596799</td>\n",
       "      <td>2.596687</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>2.501183</td>\n",
       "      <td>-0.553646</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>2.076219</td>\n",
       "      <td>0.279850</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>9</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.120052</td>\n",
       "      <td>-0.468884</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.094013</td>\n",
       "      <td>-0.991585</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>8</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.159424</td>\n",
       "      <td>-0.327614</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>9</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1.803950</td>\n",
       "      <td>-0.313487</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>7</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>2.154962</td>\n",
       "      <td>1.932715</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.525434</td>\n",
       "      <td>-0.723171</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>10</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1.839156</td>\n",
       "      <td>-1.175237</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>3.001557</td>\n",
       "      <td>-0.002691</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>5</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.508977</td>\n",
       "      <td>-0.115708</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>10</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>-0.512519</td>\n",
       "      <td>-0.567774</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>30 rows × 21 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "   age  hypertension  heart_disease  avg_glucose_level       bmi  \\\n",
       "0    8             0              1           2.500974  0.859059   \n",
       "1   10             0              1          -0.056516  0.279850   \n",
       "2    6             0              0           1.303993  0.548264   \n",
       "3   10             1              0           1.364196 -0.920950   \n",
       "4   10             0              0           1.616050 -0.214597   \n",
       "5    9             1              1          -0.802911 -0.440630   \n",
       "6    9             0              0          -0.296704 -1.090475   \n",
       "7   10             0              0          -1.042890 -0.892696   \n",
       "8   10             1              0          -0.587512 -0.115708   \n",
       "9    8             0              1           0.246375  0.887313   \n",
       "10   6             0              0          -0.085888 -0.454757   \n",
       "11  10             0              1           2.196834 -0.327614   \n",
       "12   6             1              0           1.224417  0.053817   \n",
       "13   8             0              1           1.728540  0.986203   \n",
       "14   9             1              0           2.346821 -0.666663   \n",
       "15   7             0              0          -0.404403  1.028584   \n",
       "16   9             0              0           1.777078 -1.146983   \n",
       "17   6             1              0           2.596799  2.596687   \n",
       "18  10             0              0           2.501183 -0.553646   \n",
       "19  10             0              1           2.076219  0.279850   \n",
       "20   9             0              0          -0.120052 -0.468884   \n",
       "21  10             0              0          -0.094013 -0.991585   \n",
       "22   8             0              0          -0.159424 -0.327614   \n",
       "23   9             0              1           1.803950 -0.313487   \n",
       "24   7             1              0           2.154962  1.932715   \n",
       "25   4             0              0          -0.525434 -0.723171   \n",
       "26  10             1              0           1.839156 -1.175237   \n",
       "27  10             0              1           3.001557 -0.002691   \n",
       "28   5             0              0          -0.508977 -0.115708   \n",
       "29  10             1              1          -0.512519 -0.567774   \n",
       "\n",
       "    work_type_Govt_job  work_type_Never_worked  work_type_Private  \\\n",
       "0                    0                       0                  1   \n",
       "1                    0                       0                  1   \n",
       "2                    0                       0                  1   \n",
       "3                    0                       0                  0   \n",
       "4                    0                       0                  1   \n",
       "5                    0                       0                  1   \n",
       "6                    0                       0                  1   \n",
       "7                    0                       0                  1   \n",
       "8                    0                       0                  1   \n",
       "9                    1                       0                  0   \n",
       "10                   0                       0                  1   \n",
       "11                   0                       0                  1   \n",
       "12                   0                       0                  0   \n",
       "13                   0                       0                  1   \n",
       "14                   0                       0                  1   \n",
       "15                   0                       0                  1   \n",
       "16                   1                       0                  0   \n",
       "17                   0                       0                  0   \n",
       "18                   0                       0                  0   \n",
       "19                   0                       0                  1   \n",
       "20                   0                       0                  1   \n",
       "21                   0                       0                  0   \n",
       "22                   0                       0                  1   \n",
       "23                   0                       0                  0   \n",
       "24                   0                       0                  1   \n",
       "25                   0                       0                  1   \n",
       "26                   0                       0                  0   \n",
       "27                   0                       0                  0   \n",
       "28                   1                       0                  0   \n",
       "29                   0                       0                  1   \n",
       "\n",
       "    work_type_Self-employed  smoking_status_Unknown  ...  \\\n",
       "0                         0                       0  ...   \n",
       "1                         0                       0  ...   \n",
       "2                         0                       0  ...   \n",
       "3                         1                       0  ...   \n",
       "4                         0                       0  ...   \n",
       "5                         0                       0  ...   \n",
       "6                         0                       0  ...   \n",
       "7                         0                       1  ...   \n",
       "8                         0                       0  ...   \n",
       "9                         0                       0  ...   \n",
       "10                        0                       0  ...   \n",
       "11                        0                       0  ...   \n",
       "12                        1                       0  ...   \n",
       "13                        0                       0  ...   \n",
       "14                        0                       0  ...   \n",
       "15                        0                       0  ...   \n",
       "16                        0                       0  ...   \n",
       "17                        1                       0  ...   \n",
       "18                        1                       0  ...   \n",
       "19                        0                       1  ...   \n",
       "20                        0                       0  ...   \n",
       "21                        1                       0  ...   \n",
       "22                        0                       0  ...   \n",
       "23                        1                       0  ...   \n",
       "24                        0                       0  ...   \n",
       "25                        0                       1  ...   \n",
       "26                        1                       0  ...   \n",
       "27                        1                       0  ...   \n",
       "28                        0                       0  ...   \n",
       "29                        0                       0  ...   \n",
       "\n",
       "    smoking_status_never smoked  smoking_status_smokes  gender_Female  \\\n",
       "0                             0                      0              0   \n",
       "1                             1                      0              0   \n",
       "2                             0                      1              1   \n",
       "3                             1                      0              1   \n",
       "4                             0                      0              0   \n",
       "5                             1                      0              0   \n",
       "6                             1                      0              1   \n",
       "7                             0                      0              1   \n",
       "8                             1                      0              1   \n",
       "9                             0                      1              1   \n",
       "10                            0                      1              1   \n",
       "11                            1                      0              1   \n",
       "12                            1                      0              1   \n",
       "13                            0                      1              0   \n",
       "14                            0                      1              0   \n",
       "15                            1                      0              1   \n",
       "16                            0                      1              1   \n",
       "17                            1                      0              1   \n",
       "18                            1                      0              1   \n",
       "19                            0                      0              0   \n",
       "20                            0                      0              0   \n",
       "21                            1                      0              0   \n",
       "22                            0                      0              1   \n",
       "23                            0                      1              0   \n",
       "24                            0                      1              0   \n",
       "25                            0                      0              0   \n",
       "26                            1                      0              1   \n",
       "27                            0                      0              0   \n",
       "28                            1                      0              0   \n",
       "29                            0                      0              1   \n",
       "\n",
       "    gender_Male  gender_Other  ever_married_No  ever_married_Yes  \\\n",
       "0             1             0                0                 1   \n",
       "1             1             0                0                 1   \n",
       "2             0             0                0                 1   \n",
       "3             0             0                0                 1   \n",
       "4             1             0                0                 1   \n",
       "5             1             0                0                 1   \n",
       "6             0             0                1                 0   \n",
       "7             0             0                0                 1   \n",
       "8             0             0                0                 1   \n",
       "9             0             0                0                 1   \n",
       "10            0             0                0                 1   \n",
       "11            0             0                0                 1   \n",
       "12            0             0                0                 1   \n",
       "13            1             0                0                 1   \n",
       "14            1             0                0                 1   \n",
       "15            0             0                1                 0   \n",
       "16            0             0                0                 1   \n",
       "17            0             0                0                 1   \n",
       "18            0             0                0                 1   \n",
       "19            1             0                0                 1   \n",
       "20            1             0                0                 1   \n",
       "21            1             0                0                 1   \n",
       "22            0             0                0                 1   \n",
       "23            1             0                0                 1   \n",
       "24            1             0                0                 1   \n",
       "25            1             0                0                 1   \n",
       "26            0             0                0                 1   \n",
       "27            1             0                0                 1   \n",
       "28            1             0                1                 0   \n",
       "29            0             0                1                 0   \n",
       "\n",
       "    Residence_type_Rural  Residence_type_Urban  stroke  \n",
       "0                      0                     1       1  \n",
       "1                      1                     0       1  \n",
       "2                      0                     1       1  \n",
       "3                      1                     0       1  \n",
       "4                      0                     1       1  \n",
       "5                      1                     0       1  \n",
       "6                      0                     1       1  \n",
       "7                      0                     1       1  \n",
       "8                      1                     0       1  \n",
       "9                      1                     0       1  \n",
       "10                     0                     1       1  \n",
       "11                     0                     1       1  \n",
       "12                     1                     0       1  \n",
       "13                     0                     1       1  \n",
       "14                     0                     1       1  \n",
       "15                     0                     1       1  \n",
       "16                     1                     0       1  \n",
       "17                     0                     1       1  \n",
       "18                     0                     1       1  \n",
       "19                     1                     0       1  \n",
       "20                     0                     1       1  \n",
       "21                     1                     0       1  \n",
       "22                     1                     0       1  \n",
       "23                     0                     1       1  \n",
       "24                     0                     1       1  \n",
       "25                     1                     0       1  \n",
       "26                     0                     1       1  \n",
       "27                     0                     1       1  \n",
       "28                     0                     1       1  \n",
       "29                     1                     0       1  \n",
       "\n",
       "[30 rows x 21 columns]"
      ]
     },
     "execution_count": 268,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "breaks = jenkspy.jenks_breaks(df['age'], nb_class=10)\n",
    "print(breaks)\n",
    "df['age'] = pd.cut(df['age'], bins=breaks, labels=[\"1\", \"2\", \"3\", \"4\", \"5\",\"6\",\"7\",\"8\",\"9\",\"10\"])\n",
    "df.head(30)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 269,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>age</th>\n",
       "      <th>hypertension</th>\n",
       "      <th>heart_disease</th>\n",
       "      <th>avg_glucose_level</th>\n",
       "      <th>bmi</th>\n",
       "      <th>work_type_Govt_job</th>\n",
       "      <th>work_type_Never_worked</th>\n",
       "      <th>work_type_Private</th>\n",
       "      <th>work_type_Self-employed</th>\n",
       "      <th>smoking_status_Unknown</th>\n",
       "      <th>...</th>\n",
       "      <th>smoking_status_never smoked</th>\n",
       "      <th>smoking_status_smokes</th>\n",
       "      <th>gender_Female</th>\n",
       "      <th>gender_Male</th>\n",
       "      <th>gender_Other</th>\n",
       "      <th>ever_married_No</th>\n",
       "      <th>ever_married_Yes</th>\n",
       "      <th>Residence_type_Rural</th>\n",
       "      <th>Residence_type_Urban</th>\n",
       "      <th>stroke</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>579</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.352949</td>\n",
       "      <td>1.268744</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>622</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.506686</td>\n",
       "      <td>0.180960</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>642</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.040267</td>\n",
       "      <td>-0.920950</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>652</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.896236</td>\n",
       "      <td>-0.765552</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>694</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.379405</td>\n",
       "      <td>-1.288253</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>697</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.511894</td>\n",
       "      <td>-0.822060</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>708</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.655007</td>\n",
       "      <td>0.972075</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>730</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.451482</td>\n",
       "      <td>-0.666663</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>763</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.216794</td>\n",
       "      <td>0.760170</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>773</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.706961</td>\n",
       "      <td>-1.090475</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>796</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.745624</td>\n",
       "      <td>-1.104602</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>806</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.197754</td>\n",
       "      <td>-0.341741</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>858</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.698961</td>\n",
       "      <td>-1.245872</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>871</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.262748</td>\n",
       "      <td>-1.217618</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>895</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.043184</td>\n",
       "      <td>-0.087454</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>923</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.245666</td>\n",
       "      <td>-0.045072</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1254</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.553349</td>\n",
       "      <td>-0.214597</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1371</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.043059</td>\n",
       "      <td>1.127473</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1377</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.703544</td>\n",
       "      <td>-0.299359</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1521</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.470731</td>\n",
       "      <td>-1.330634</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1649</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.735416</td>\n",
       "      <td>-1.062220</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1741</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.752290</td>\n",
       "      <td>0.209214</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1860</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.910610</td>\n",
       "      <td>-1.132856</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1898</th>\n",
       "      <td>NaN</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.888945</td>\n",
       "      <td>-0.214597</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1902</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.525018</td>\n",
       "      <td>1.113346</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1949</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.519685</td>\n",
       "      <td>-0.920950</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2177</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.600220</td>\n",
       "      <td>-0.624282</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2329</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.098179</td>\n",
       "      <td>-0.906823</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2408</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.246375</td>\n",
       "      <td>-1.175237</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2649</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.612094</td>\n",
       "      <td>-0.977458</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2868</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.311994</td>\n",
       "      <td>-0.751425</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2982</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.228043</td>\n",
       "      <td>-0.807933</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2983</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.892486</td>\n",
       "      <td>-0.271105</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2999</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.667922</td>\n",
       "      <td>-0.553646</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3043</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-1.078720</td>\n",
       "      <td>-1.330634</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3311</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-1.089553</td>\n",
       "      <td>-0.158089</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3339</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.467523</td>\n",
       "      <td>-0.610155</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3471</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.349908</td>\n",
       "      <td>0.237469</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3511</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-1.067263</td>\n",
       "      <td>-1.076347</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3539</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.585221</td>\n",
       "      <td>-1.697938</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3623</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.435650</td>\n",
       "      <td>-1.245872</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3629</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.402736</td>\n",
       "      <td>-1.189364</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3680</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.604802</td>\n",
       "      <td>0.873186</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3718</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.873530</td>\n",
       "      <td>-0.822060</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3788</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.245458</td>\n",
       "      <td>-1.005712</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3808</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.280247</td>\n",
       "      <td>-0.793806</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3881</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-1.046640</td>\n",
       "      <td>-0.327614</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3912</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.167340</td>\n",
       "      <td>-1.415397</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3996</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.376280</td>\n",
       "      <td>-0.892696</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4113</th>\n",
       "      <td>NaN</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.754998</td>\n",
       "      <td>-0.138853</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>50 rows × 21 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      age  hypertension  heart_disease  avg_glucose_level       bmi  \\\n",
       "579   NaN             0              0          -0.352949  1.268744   \n",
       "622   NaN             0              0          -0.506686  0.180960   \n",
       "642   NaN             0              0          -0.040267 -0.920950   \n",
       "652   NaN             0              0          -0.896236 -0.765552   \n",
       "694   NaN             0              0          -0.379405 -1.288253   \n",
       "697   NaN             0              0          -0.511894 -0.822060   \n",
       "708   NaN             0              0          -0.655007  0.972075   \n",
       "730   NaN             0              0          -0.451482 -0.666663   \n",
       "763   NaN             0              0           0.216794  0.760170   \n",
       "773   NaN             0              0           0.706961 -1.090475   \n",
       "796   NaN             0              0          -0.745624 -1.104602   \n",
       "806   NaN             0              0          -0.197754 -0.341741   \n",
       "858   NaN             0              0          -0.698961 -1.245872   \n",
       "871   NaN             0              0          -0.262748 -1.217618   \n",
       "895   NaN             0              0          -0.043184 -0.087454   \n",
       "923   NaN             0              0          -0.245666 -0.045072   \n",
       "1254  NaN             0              0          -0.553349 -0.214597   \n",
       "1371  NaN             0              0           0.043059  1.127473   \n",
       "1377  NaN             0              0          -0.703544 -0.299359   \n",
       "1521  NaN             0              0           0.470731 -1.330634   \n",
       "1649  NaN             0              0          -0.735416 -1.062220   \n",
       "1741  NaN             0              0          -0.752290  0.209214   \n",
       "1860  NaN             0              0          -0.910610 -1.132856   \n",
       "1898  NaN             1              0          -0.888945 -0.214597   \n",
       "1902  NaN             0              0          -0.525018  1.113346   \n",
       "1949  NaN             0              0           0.519685 -0.920950   \n",
       "2177  NaN             0              0          -0.600220 -0.624282   \n",
       "2329  NaN             0              0          -0.098179 -0.906823   \n",
       "2408  NaN             0              0           0.246375 -1.175237   \n",
       "2649  NaN             0              0          -0.612094 -0.977458   \n",
       "2868  NaN             0              0           0.311994 -0.751425   \n",
       "2982  NaN             0              0           0.228043 -0.807933   \n",
       "2983  NaN             0              0          -0.892486 -0.271105   \n",
       "2999  NaN             0              0          -0.667922 -0.553646   \n",
       "3043  NaN             0              0          -1.078720 -1.330634   \n",
       "3311  NaN             0              0          -1.089553 -0.158089   \n",
       "3339  NaN             0              0          -0.467523 -0.610155   \n",
       "3471  NaN             0              0           0.349908  0.237469   \n",
       "3511  NaN             0              0          -1.067263 -1.076347   \n",
       "3539  NaN             0              0          -0.585221 -1.697938   \n",
       "3623  NaN             0              0          -0.435650 -1.245872   \n",
       "3629  NaN             0              0          -0.402736 -1.189364   \n",
       "3680  NaN             0              0          -0.604802  0.873186   \n",
       "3718  NaN             0              0          -0.873530 -0.822060   \n",
       "3788  NaN             0              0          -0.245458 -1.005712   \n",
       "3808  NaN             0              0          -0.280247 -0.793806   \n",
       "3881  NaN             0              0          -1.046640 -0.327614   \n",
       "3912  NaN             0              0          -0.167340 -1.415397   \n",
       "3996  NaN             0              0          -0.376280 -0.892696   \n",
       "4113  NaN             0              0          -0.754998 -0.138853   \n",
       "\n",
       "      work_type_Govt_job  work_type_Never_worked  work_type_Private  \\\n",
       "579                    0                       0                  1   \n",
       "622                    0                       0                  1   \n",
       "642                    0                       0                  1   \n",
       "652                    0                       0                  1   \n",
       "694                    0                       0                  1   \n",
       "697                    0                       0                  0   \n",
       "708                    0                       0                  1   \n",
       "730                    0                       1                  0   \n",
       "763                    0                       0                  1   \n",
       "773                    0                       0                  1   \n",
       "796                    0                       0                  1   \n",
       "806                    0                       0                  1   \n",
       "858                    0                       0                  1   \n",
       "871                    0                       0                  1   \n",
       "895                    0                       0                  1   \n",
       "923                    0                       0                  1   \n",
       "1254                   0                       0                  0   \n",
       "1371                   0                       0                  1   \n",
       "1377                   0                       0                  1   \n",
       "1521                   0                       0                  1   \n",
       "1649                   0                       0                  1   \n",
       "1741                   0                       0                  1   \n",
       "1860                   0                       0                  1   \n",
       "1898                   0                       0                  1   \n",
       "1902                   0                       0                  1   \n",
       "1949                   1                       0                  0   \n",
       "2177                   0                       0                  1   \n",
       "2329                   0                       0                  1   \n",
       "2408                   0                       0                  1   \n",
       "2649                   0                       0                  1   \n",
       "2868                   0                       0                  1   \n",
       "2982                   0                       0                  1   \n",
       "2983                   0                       0                  1   \n",
       "2999                   0                       0                  1   \n",
       "3043                   0                       0                  1   \n",
       "3311                   0                       0                  1   \n",
       "3339                   0                       0                  1   \n",
       "3471                   0                       0                  1   \n",
       "3511                   0                       0                  1   \n",
       "3539                   0                       0                  1   \n",
       "3623                   0                       0                  1   \n",
       "3629                   0                       0                  1   \n",
       "3680                   0                       0                  1   \n",
       "3718                   0                       0                  1   \n",
       "3788                   0                       0                  1   \n",
       "3808                   0                       0                  1   \n",
       "3881                   0                       0                  1   \n",
       "3912                   0                       0                  1   \n",
       "3996                   0                       0                  1   \n",
       "4113                   0                       0                  1   \n",
       "\n",
       "      work_type_Self-employed  smoking_status_Unknown  ...  \\\n",
       "579                         0                       1  ...   \n",
       "622                         0                       0  ...   \n",
       "642                         0                       0  ...   \n",
       "652                         0                       1  ...   \n",
       "694                         0                       0  ...   \n",
       "697                         1                       0  ...   \n",
       "708                         0                       0  ...   \n",
       "730                         0                       0  ...   \n",
       "763                         0                       1  ...   \n",
       "773                         0                       1  ...   \n",
       "796                         0                       0  ...   \n",
       "806                         0                       0  ...   \n",
       "858                         0                       1  ...   \n",
       "871                         0                       0  ...   \n",
       "895                         0                       0  ...   \n",
       "923                         0                       0  ...   \n",
       "1254                        1                       0  ...   \n",
       "1371                        0                       0  ...   \n",
       "1377                        0                       0  ...   \n",
       "1521                        0                       1  ...   \n",
       "1649                        0                       0  ...   \n",
       "1741                        0                       1  ...   \n",
       "1860                        0                       1  ...   \n",
       "1898                        0                       0  ...   \n",
       "1902                        0                       1  ...   \n",
       "1949                        0                       0  ...   \n",
       "2177                        0                       1  ...   \n",
       "2329                        0                       1  ...   \n",
       "2408                        0                       1  ...   \n",
       "2649                        0                       1  ...   \n",
       "2868                        0                       1  ...   \n",
       "2982                        0                       1  ...   \n",
       "2983                        0                       0  ...   \n",
       "2999                        0                       1  ...   \n",
       "3043                        0                       0  ...   \n",
       "3311                        0                       1  ...   \n",
       "3339                        0                       0  ...   \n",
       "3471                        0                       1  ...   \n",
       "3511                        0                       1  ...   \n",
       "3539                        0                       0  ...   \n",
       "3623                        0                       0  ...   \n",
       "3629                        0                       0  ...   \n",
       "3680                        0                       1  ...   \n",
       "3718                        0                       0  ...   \n",
       "3788                        0                       1  ...   \n",
       "3808                        0                       0  ...   \n",
       "3881                        0                       0  ...   \n",
       "3912                        0                       0  ...   \n",
       "3996                        0                       1  ...   \n",
       "4113                        0                       0  ...   \n",
       "\n",
       "      smoking_status_never smoked  smoking_status_smokes  gender_Female  \\\n",
       "579                             0                      0              1   \n",
       "622                             1                      0              0   \n",
       "642                             1                      0              0   \n",
       "652                             0                      0              0   \n",
       "694                             1                      0              1   \n",
       "697                             1                      0              1   \n",
       "708                             0                      1              1   \n",
       "730                             1                      0              1   \n",
       "763                             0                      0              0   \n",
       "773                             0                      0              0   \n",
       "796                             1                      0              1   \n",
       "806                             1                      0              0   \n",
       "858                             0                      0              1   \n",
       "871                             1                      0              1   \n",
       "895                             1                      0              1   \n",
       "923                             0                      0              0   \n",
       "1254                            1                      0              0   \n",
       "1371                            1                      0              1   \n",
       "1377                            1                      0              0   \n",
       "1521                            0                      0              1   \n",
       "1649                            1                      0              0   \n",
       "1741                            0                      0              1   \n",
       "1860                            0                      0              0   \n",
       "1898                            1                      0              1   \n",
       "1902                            0                      0              1   \n",
       "1949                            1                      0              1   \n",
       "2177                            0                      0              0   \n",
       "2329                            0                      0              0   \n",
       "2408                            0                      0              0   \n",
       "2649                            0                      0              1   \n",
       "2868                            0                      0              0   \n",
       "2982                            0                      0              0   \n",
       "2983                            0                      1              1   \n",
       "2999                            0                      0              1   \n",
       "3043                            1                      0              1   \n",
       "3311                            0                      0              0   \n",
       "3339                            1                      0              0   \n",
       "3471                            0                      0              1   \n",
       "3511                            0                      0              1   \n",
       "3539                            0                      1              0   \n",
       "3623                            1                      0              1   \n",
       "3629                            1                      0              1   \n",
       "3680                            0                      0              0   \n",
       "3718                            1                      0              1   \n",
       "3788                            0                      0              1   \n",
       "3808                            0                      1              0   \n",
       "3881                            1                      0              1   \n",
       "3912                            1                      0              1   \n",
       "3996                            0                      0              1   \n",
       "4113                            0                      1              1   \n",
       "\n",
       "      gender_Male  gender_Other  ever_married_No  ever_married_Yes  \\\n",
       "579             0             0                1                 0   \n",
       "622             1             0                1                 0   \n",
       "642             1             0                1                 0   \n",
       "652             1             0                1                 0   \n",
       "694             0             0                0                 1   \n",
       "697             0             0                1                 0   \n",
       "708             0             0                1                 0   \n",
       "730             0             0                1                 0   \n",
       "763             1             0                1                 0   \n",
       "773             1             0                1                 0   \n",
       "796             0             0                1                 0   \n",
       "806             1             0                0                 1   \n",
       "858             0             0                1                 0   \n",
       "871             0             0                1                 0   \n",
       "895             0             0                1                 0   \n",
       "923             1             0                1                 0   \n",
       "1254            1             0                1                 0   \n",
       "1371            0             0                1                 0   \n",
       "1377            1             0                1                 0   \n",
       "1521            0             0                1                 0   \n",
       "1649            1             0                1                 0   \n",
       "1741            0             0                1                 0   \n",
       "1860            1             0                1                 0   \n",
       "1898            0             0                1                 0   \n",
       "1902            0             0                1                 0   \n",
       "1949            0             0                1                 0   \n",
       "2177            1             0                1                 0   \n",
       "2329            1             0                1                 0   \n",
       "2408            1             0                1                 0   \n",
       "2649            0             0                1                 0   \n",
       "2868            1             0                1                 0   \n",
       "2982            1             0                1                 0   \n",
       "2983            0             0                1                 0   \n",
       "2999            0             0                1                 0   \n",
       "3043            0             0                1                 0   \n",
       "3311            1             0                1                 0   \n",
       "3339            1             0                1                 0   \n",
       "3471            0             0                1                 0   \n",
       "3511            0             0                1                 0   \n",
       "3539            1             0                1                 0   \n",
       "3623            0             0                1                 0   \n",
       "3629            0             0                1                 0   \n",
       "3680            1             0                1                 0   \n",
       "3718            0             0                1                 0   \n",
       "3788            0             0                1                 0   \n",
       "3808            1             0                1                 0   \n",
       "3881            0             0                1                 0   \n",
       "3912            0             0                1                 0   \n",
       "3996            0             0                1                 0   \n",
       "4113            0             0                1                 0   \n",
       "\n",
       "      Residence_type_Rural  Residence_type_Urban  stroke  \n",
       "579                      1                     0       0  \n",
       "622                      1                     0       0  \n",
       "642                      1                     0       0  \n",
       "652                      0                     1       0  \n",
       "694                      1                     0       0  \n",
       "697                      0                     1       0  \n",
       "708                      1                     0       0  \n",
       "730                      0                     1       0  \n",
       "763                      1                     0       0  \n",
       "773                      1                     0       0  \n",
       "796                      1                     0       0  \n",
       "806                      0                     1       0  \n",
       "858                      1                     0       0  \n",
       "871                      1                     0       0  \n",
       "895                      1                     0       0  \n",
       "923                      1                     0       0  \n",
       "1254                     1                     0       0  \n",
       "1371                     0                     1       0  \n",
       "1377                     0                     1       0  \n",
       "1521                     0                     1       0  \n",
       "1649                     0                     1       0  \n",
       "1741                     1                     0       0  \n",
       "1860                     1                     0       0  \n",
       "1898                     1                     0       0  \n",
       "1902                     0                     1       0  \n",
       "1949                     0                     1       0  \n",
       "2177                     1                     0       0  \n",
       "2329                     0                     1       0  \n",
       "2408                     1                     0       0  \n",
       "2649                     0                     1       0  \n",
       "2868                     0                     1       0  \n",
       "2982                     0                     1       0  \n",
       "2983                     0                     1       0  \n",
       "2999                     0                     1       0  \n",
       "3043                     1                     0       0  \n",
       "3311                     1                     0       0  \n",
       "3339                     1                     0       0  \n",
       "3471                     0                     1       0  \n",
       "3511                     1                     0       0  \n",
       "3539                     0                     1       0  \n",
       "3623                     0                     1       0  \n",
       "3629                     0                     1       0  \n",
       "3680                     1                     0       0  \n",
       "3718                     1                     0       0  \n",
       "3788                     0                     1       0  \n",
       "3808                     1                     0       0  \n",
       "3881                     0                     1       0  \n",
       "3912                     0                     1       0  \n",
       "3996                     0                     1       0  \n",
       "4113                     1                     0       0  \n",
       "\n",
       "[50 rows x 21 columns]"
      ]
     },
     "execution_count": 269,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df[df['age'].isna()==True]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 270,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [],
   "source": [
    "df['age'].fillna(\"1\",inplace=True)\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 271,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[-1.1147589901189745, -0.6429242284489848, -0.2056698510779877, 0.3982369807459855, 1.303993070697912, 2.185376239186091, 3.3977728453314957]\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>age</th>\n",
       "      <th>hypertension</th>\n",
       "      <th>heart_disease</th>\n",
       "      <th>avg_glucose_level</th>\n",
       "      <th>bmi</th>\n",
       "      <th>work_type_Govt_job</th>\n",
       "      <th>work_type_Never_worked</th>\n",
       "      <th>work_type_Private</th>\n",
       "      <th>work_type_Self-employed</th>\n",
       "      <th>smoking_status_Unknown</th>\n",
       "      <th>...</th>\n",
       "      <th>smoking_status_never smoked</th>\n",
       "      <th>smoking_status_smokes</th>\n",
       "      <th>gender_Female</th>\n",
       "      <th>gender_Male</th>\n",
       "      <th>gender_Other</th>\n",
       "      <th>ever_married_No</th>\n",
       "      <th>ever_married_Yes</th>\n",
       "      <th>Residence_type_Rural</th>\n",
       "      <th>Residence_type_Urban</th>\n",
       "      <th>stroke</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>8</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>6</td>\n",
       "      <td>0.859059</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>3</td>\n",
       "      <td>0.279850</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>6</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>4</td>\n",
       "      <td>0.548264</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>10</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>-0.920950</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>10</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>-0.214597</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4189</th>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>-0.227499</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4190</th>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "      <td>0.345098</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4191</th>\n",
       "      <td>5</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0.402209</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4192</th>\n",
       "      <td>4</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0.025377</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4193</th>\n",
       "      <td>10</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>0.179082</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>4194 rows × 21 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "     age  hypertension  heart_disease avg_glucose_level       bmi  \\\n",
       "0      8             0              1                 6  0.859059   \n",
       "1     10             0              1                 3  0.279850   \n",
       "2      6             0              0                 4  0.548264   \n",
       "3     10             1              0                 5 -0.920950   \n",
       "4     10             0              0                 5 -0.214597   \n",
       "...   ..           ...            ...               ...       ...   \n",
       "4189   4             0              0                 1 -0.227499   \n",
       "4190   4             0              0                 5  0.345098   \n",
       "4191   5             1              0                 2  0.402209   \n",
       "4192   4             0              0                 2  0.025377   \n",
       "4193  10             1              0                 2  0.179082   \n",
       "\n",
       "      work_type_Govt_job  work_type_Never_worked  work_type_Private  \\\n",
       "0                      0                       0                  1   \n",
       "1                      0                       0                  1   \n",
       "2                      0                       0                  1   \n",
       "3                      0                       0                  0   \n",
       "4                      0                       0                  1   \n",
       "...                  ...                     ...                ...   \n",
       "4189                   0                       0                  1   \n",
       "4190                   0                       0                  1   \n",
       "4191                   1                       0                  0   \n",
       "4192                   0                       0                  1   \n",
       "4193                   0                       0                  1   \n",
       "\n",
       "      work_type_Self-employed  smoking_status_Unknown  ...  \\\n",
       "0                           0                       0  ...   \n",
       "1                           0                       0  ...   \n",
       "2                           0                       0  ...   \n",
       "3                           1                       0  ...   \n",
       "4                           0                       0  ...   \n",
       "...                       ...                     ...  ...   \n",
       "4189                        0                       0  ...   \n",
       "4190                        0                       0  ...   \n",
       "4191                        0                       0  ...   \n",
       "4192                        0                       0  ...   \n",
       "4193                        0                       0  ...   \n",
       "\n",
       "      smoking_status_never smoked  smoking_status_smokes  gender_Female  \\\n",
       "0                               0                      0              0   \n",
       "1                               1                      0              0   \n",
       "2                               0                      1              1   \n",
       "3                               1                      0              1   \n",
       "4                               0                      0              0   \n",
       "...                           ...                    ...            ...   \n",
       "4189                            0                      0              0   \n",
       "4190                            0                      1              0   \n",
       "4191                            0                      1              1   \n",
       "4192                            0                      1              0   \n",
       "4193                            1                      0              1   \n",
       "\n",
       "      gender_Male  gender_Other  ever_married_No  ever_married_Yes  \\\n",
       "0               1             0                0                 1   \n",
       "1               1             0                0                 1   \n",
       "2               0             0                0                 1   \n",
       "3               0             0                0                 1   \n",
       "4               1             0                0                 1   \n",
       "...           ...           ...              ...               ...   \n",
       "4189            1             0                1                 0   \n",
       "4190            1             0                0                 1   \n",
       "4191            0             0                0                 1   \n",
       "4192            1             0                0                 1   \n",
       "4193            0             0                0                 1   \n",
       "\n",
       "      Residence_type_Rural  Residence_type_Urban  stroke  \n",
       "0                        0                     1       1  \n",
       "1                        1                     0       1  \n",
       "2                        0                     1       1  \n",
       "3                        1                     0       1  \n",
       "4                        0                     1       1  \n",
       "...                    ...                   ...     ...  \n",
       "4189                     1                     0       0  \n",
       "4190                     0                     1       0  \n",
       "4191                     1                     0       0  \n",
       "4192                     1                     0       0  \n",
       "4193                     0                     1       0  \n",
       "\n",
       "[4194 rows x 21 columns]"
      ]
     },
     "execution_count": 271,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "breaks = jenkspy.jenks_breaks(df['avg_glucose_level'], nb_class=6)\n",
    "print(breaks)\n",
    "df['avg_glucose_level'] = pd.cut(df['avg_glucose_level'], bins=breaks, labels=[\"1\", \"2\", \"3\", \"4\", \"5\", \"6\"])\n",
    "df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 272,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "age                               0\n",
       "hypertension                      0\n",
       "heart_disease                     0\n",
       "avg_glucose_level                 0\n",
       "bmi                               0\n",
       "work_type_Govt_job                0\n",
       "work_type_Never_worked            0\n",
       "work_type_Private                 0\n",
       "work_type_Self-employed           0\n",
       "smoking_status_Unknown            0\n",
       "smoking_status_formerly smoked    0\n",
       "smoking_status_never smoked       0\n",
       "smoking_status_smokes             0\n",
       "gender_Female                     0\n",
       "gender_Male                       0\n",
       "gender_Other                      0\n",
       "ever_married_No                   0\n",
       "ever_married_Yes                  0\n",
       "Residence_type_Rural              0\n",
       "Residence_type_Urban              0\n",
       "stroke                            0\n",
       "dtype: int64"
      ]
     },
     "execution_count": 272,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df['avg_glucose_level'].fillna(\"1\",inplace=True)\n",
    "df.isna().sum()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 273,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>hypertension</th>\n",
       "      <th>heart_disease</th>\n",
       "      <th>bmi</th>\n",
       "      <th>work_type_Govt_job</th>\n",
       "      <th>work_type_Never_worked</th>\n",
       "      <th>work_type_Private</th>\n",
       "      <th>work_type_Self-employed</th>\n",
       "      <th>smoking_status_Unknown</th>\n",
       "      <th>smoking_status_formerly smoked</th>\n",
       "      <th>smoking_status_never smoked</th>\n",
       "      <th>...</th>\n",
       "      <th>age_6</th>\n",
       "      <th>age_7</th>\n",
       "      <th>age_8</th>\n",
       "      <th>age_9</th>\n",
       "      <th>age_10</th>\n",
       "      <th>avg_glucose_level_2</th>\n",
       "      <th>avg_glucose_level_3</th>\n",
       "      <th>avg_glucose_level_4</th>\n",
       "      <th>avg_glucose_level_5</th>\n",
       "      <th>avg_glucose_level_6</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0.859059</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0.279850</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.548264</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.920950</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.214597</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4189</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.227499</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4190</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.345098</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4191</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0.402209</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4192</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.025377</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4193</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0.179082</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>4194 rows × 33 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      hypertension  heart_disease       bmi  work_type_Govt_job  \\\n",
       "0                0              1  0.859059                   0   \n",
       "1                0              1  0.279850                   0   \n",
       "2                0              0  0.548264                   0   \n",
       "3                1              0 -0.920950                   0   \n",
       "4                0              0 -0.214597                   0   \n",
       "...            ...            ...       ...                 ...   \n",
       "4189             0              0 -0.227499                   0   \n",
       "4190             0              0  0.345098                   0   \n",
       "4191             1              0  0.402209                   1   \n",
       "4192             0              0  0.025377                   0   \n",
       "4193             1              0  0.179082                   0   \n",
       "\n",
       "      work_type_Never_worked  work_type_Private  work_type_Self-employed  \\\n",
       "0                          0                  1                        0   \n",
       "1                          0                  1                        0   \n",
       "2                          0                  1                        0   \n",
       "3                          0                  0                        1   \n",
       "4                          0                  1                        0   \n",
       "...                      ...                ...                      ...   \n",
       "4189                       0                  1                        0   \n",
       "4190                       0                  1                        0   \n",
       "4191                       0                  0                        0   \n",
       "4192                       0                  1                        0   \n",
       "4193                       0                  1                        0   \n",
       "\n",
       "      smoking_status_Unknown  smoking_status_formerly smoked  \\\n",
       "0                          0                               1   \n",
       "1                          0                               0   \n",
       "2                          0                               0   \n",
       "3                          0                               0   \n",
       "4                          0                               1   \n",
       "...                      ...                             ...   \n",
       "4189                       0                               1   \n",
       "4190                       0                               0   \n",
       "4191                       0                               0   \n",
       "4192                       0                               0   \n",
       "4193                       0                               0   \n",
       "\n",
       "      smoking_status_never smoked  ...  age_6  age_7  age_8  age_9  age_10  \\\n",
       "0                               0  ...      0      0      1      0       0   \n",
       "1                               1  ...      0      0      0      0       1   \n",
       "2                               0  ...      1      0      0      0       0   \n",
       "3                               1  ...      0      0      0      0       1   \n",
       "4                               0  ...      0      0      0      0       1   \n",
       "...                           ...  ...    ...    ...    ...    ...     ...   \n",
       "4189                            0  ...      0      0      0      0       0   \n",
       "4190                            0  ...      0      0      0      0       0   \n",
       "4191                            0  ...      0      0      0      0       0   \n",
       "4192                            0  ...      0      0      0      0       0   \n",
       "4193                            1  ...      0      0      0      0       1   \n",
       "\n",
       "      avg_glucose_level_2  avg_glucose_level_3  avg_glucose_level_4  \\\n",
       "0                       0                    0                    0   \n",
       "1                       0                    1                    0   \n",
       "2                       0                    0                    1   \n",
       "3                       0                    0                    0   \n",
       "4                       0                    0                    0   \n",
       "...                   ...                  ...                  ...   \n",
       "4189                    0                    0                    0   \n",
       "4190                    0                    0                    0   \n",
       "4191                    1                    0                    0   \n",
       "4192                    1                    0                    0   \n",
       "4193                    1                    0                    0   \n",
       "\n",
       "      avg_glucose_level_5  avg_glucose_level_6  \n",
       "0                       0                    1  \n",
       "1                       0                    0  \n",
       "2                       0                    0  \n",
       "3                       1                    0  \n",
       "4                       1                    0  \n",
       "...                   ...                  ...  \n",
       "4189                    0                    0  \n",
       "4190                    1                    0  \n",
       "4191                    0                    0  \n",
       "4192                    0                    0  \n",
       "4193                    0                    0  \n",
       "\n",
       "[4194 rows x 33 columns]"
      ]
     },
     "execution_count": 273,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df=pd.get_dummies(df,columns=['age','avg_glucose_level'])\n",
    "df=df.drop(columns=['age_1','avg_glucose_level_1'])\n",
    "df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 274,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>hypertension</th>\n",
       "      <th>heart_disease</th>\n",
       "      <th>bmi</th>\n",
       "      <th>work_type_Govt_job</th>\n",
       "      <th>work_type_Never_worked</th>\n",
       "      <th>work_type_Private</th>\n",
       "      <th>work_type_Self-employed</th>\n",
       "      <th>smoking_status_Unknown</th>\n",
       "      <th>smoking_status_formerly smoked</th>\n",
       "      <th>smoking_status_never smoked</th>\n",
       "      <th>...</th>\n",
       "      <th>age_6</th>\n",
       "      <th>age_7</th>\n",
       "      <th>age_8</th>\n",
       "      <th>age_9</th>\n",
       "      <th>age_10</th>\n",
       "      <th>avg_glucose_level_2</th>\n",
       "      <th>avg_glucose_level_3</th>\n",
       "      <th>avg_glucose_level_4</th>\n",
       "      <th>avg_glucose_level_5</th>\n",
       "      <th>avg_glucose_level_6</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>759</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.336358</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1359</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.689534</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2516</th>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0.039690</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1179</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0.618899</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1147</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0.392866</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2825</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.186343</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2046</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.709044</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2112</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-1.570794</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3219</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.322231</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>272</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.878569</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>3355 rows × 33 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      hypertension  heart_disease       bmi  work_type_Govt_job  \\\n",
       "759              0              0  0.336358                   0   \n",
       "1359             0              0  0.689534                   0   \n",
       "2516             0              1  0.039690                   1   \n",
       "1179             1              0  0.618899                   0   \n",
       "1147             1              0  0.392866                   0   \n",
       "...            ...            ...       ...                 ...   \n",
       "2825             0              0 -0.186343                   1   \n",
       "2046             0              0 -0.709044                   0   \n",
       "2112             0              0 -1.570794                   0   \n",
       "3219             0              0  0.322231                   1   \n",
       "272              0              0 -0.878569                   1   \n",
       "\n",
       "      work_type_Never_worked  work_type_Private  work_type_Self-employed  \\\n",
       "759                        0                  1                        0   \n",
       "1359                       0                  1                        0   \n",
       "2516                       0                  0                        0   \n",
       "1179                       0                  0                        1   \n",
       "1147                       0                  1                        0   \n",
       "...                      ...                ...                      ...   \n",
       "2825                       0                  0                        0   \n",
       "2046                       0                  1                        0   \n",
       "2112                       0                  1                        0   \n",
       "3219                       0                  0                        0   \n",
       "272                        0                  0                        0   \n",
       "\n",
       "      smoking_status_Unknown  smoking_status_formerly smoked  \\\n",
       "759                        1                               0   \n",
       "1359                       0                               0   \n",
       "2516                       0                               0   \n",
       "1179                       0                               0   \n",
       "1147                       0                               0   \n",
       "...                      ...                             ...   \n",
       "2825                       0                               0   \n",
       "2046                       0                               0   \n",
       "2112                       0                               0   \n",
       "3219                       1                               0   \n",
       "272                        0                               1   \n",
       "\n",
       "      smoking_status_never smoked  ...  age_6  age_7  age_8  age_9  age_10  \\\n",
       "759                             0  ...      0      0      0      0       1   \n",
       "1359                            0  ...      0      0      0      0       1   \n",
       "2516                            0  ...      1      0      0      0       0   \n",
       "1179                            1  ...      1      0      0      0       0   \n",
       "1147                            1  ...      0      0      0      0       1   \n",
       "...                           ...  ...    ...    ...    ...    ...     ...   \n",
       "2825                            1  ...      1      0      0      0       0   \n",
       "2046                            1  ...      0      0      0      0       0   \n",
       "2112                            1  ...      0      0      0      0       0   \n",
       "3219                            0  ...      0      0      0      0       0   \n",
       "272                             0  ...      0      0      0      0       0   \n",
       "\n",
       "      avg_glucose_level_2  avg_glucose_level_3  avg_glucose_level_4  \\\n",
       "759                     0                    1                    0   \n",
       "1359                    0                    0                    1   \n",
       "2516                    1                    0                    0   \n",
       "1179                    0                    0                    0   \n",
       "1147                    0                    0                    0   \n",
       "...                   ...                  ...                  ...   \n",
       "2825                    1                    0                    0   \n",
       "2046                    1                    0                    0   \n",
       "2112                    1                    0                    0   \n",
       "3219                    0                    0                    0   \n",
       "272                     1                    0                    0   \n",
       "\n",
       "      avg_glucose_level_5  avg_glucose_level_6  \n",
       "759                     0                    0  \n",
       "1359                    0                    0  \n",
       "2516                    0                    0  \n",
       "1179                    0                    0  \n",
       "1147                    0                    1  \n",
       "...                   ...                  ...  \n",
       "2825                    0                    0  \n",
       "2046                    0                    0  \n",
       "2112                    0                    0  \n",
       "3219                    0                    0  \n",
       "272                     0                    0  \n",
       "\n",
       "[3355 rows x 33 columns]"
      ]
     },
     "execution_count": 274,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "train, test = train_test_split(df,test_size=0.2)\n",
    "train"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### stroke = 1 <<<< stroke =0 => need to add"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 275,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<AxesSubplot:title={'center':'count (Stroke)'}>"
      ]
     },
     "execution_count": 275,
     "metadata": {},
     "output_type": "execute_result"
    },
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAX0AAAEFCAYAAAAPCDf9AAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjQuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/Z1A+gAAAACXBIWXMAAAsTAAALEwEAmpwYAAASyUlEQVR4nO3df5Cd1X3f8ffHEuAkYISjrQb0w1JtpSmkMfYoMhl3aic0/PI4ODONK6aDZYaO0hYycZv+EB53cJ3Q2m1sZpw4THHRWHYTsCa2a8WVQxTiDCGpDSKmGEEJW4EsKTISP43NGEf42z/u2eYidrW70uqu0Hm/Zu7c5/me85x7Hs3qcx+d+9xVqgpJUh9eNd8TkCSNjqEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+aoSSnJXkwydkjer33JrnrKI57Z5LPHo856ZXP0JeAJI8l+YfTdNsA3FlV+9sxy5J8LskTSZ5N8kCS97a2lUkqycLjPPWXqarfB85L8pOjfm2d+Ax9aeb+GfCZof3PAHuA1wE/ClwJPD7TwY7zG8KtDN6kpJcw9HXCSbI8yeeTHEzyZJLfavVXJflAkt1JDiT5dJIzW9vbk+w9bJz/f/We5INJtrRjnkuyM8ma1vYZYAXw+0m+k+TfTjKnFcDfBr42VP4p4FNV9d2qOlRVX6+qL7e2O9vzM23Mn27LNX+W5MYkTwIfTHJmm9PBdl4fSDLp38sk/yXJXe2YM5PckmR/kn1Jfj3JgqHufwK8YzZ/7uqDoa8TSguuLwG7gZXAUuC21vze9vgZBgF8OvBbsxj+59tYi4CtE8dW1ZXAN4F3VtXpVfWfJzn27wG7qurQUO2rwCeSrGtvCsP+QXte1Mb8X23/LcAuYAlwA/CbwJntfN4GvAe4anig9mb3SeAngYuq6lngU8Ah4A3Am4CLgH86dNhDwMokr5nuD0V9MfR1olkLnAP8m3YF/b2qmvgw858AH6uqXVX1HeA6YN0slknuqqptVfUig6WZN85iXouA5w6r/SLwp8C/Bx5Ncl+Sn5pmnL+qqt9sbx7fB9YB11XVc1X1GPBRBstEE05hsFTzWgZvSs8nWQJcBryv/RkdAG5sY02YmOuiWZyjOjDyD5mkaSwHdh92RT3hHAb/Apiwm8HP8JIZjv2toe3ngVcnWTjFax3uaeCM4UJVPQ1sBDYmWQz8BvA/kiw7wjh7hrYXMwj1w89p6dD+Gxi8Oa2tqu+32uvacfuTTPR71WFjT8z1mSOelbrjlb5ONHuAFVNcvf8Vg8CbsILBEsfjwHeBH55oaMtEY7N43el+3ez9wKqp/lVRVU8wCP1zGFyVTzXecP0J4K95+TntG9p/iMFyz5eT/J1W2wO8ACyuqkXt8ZqqOm/ouL8LPFZV357mvNQZQ18nmruB/cCHk/xIklcneWtruxX4l0lWJTkd+I/AZ9uV+l8yuHJ/R5JTgA8Ap83idR9nsK4+qaraC4wzWH4CIMlHkvxEkoVJzgD+OTBeVU8CB4EfTDPmi8AW4IYkZyR5HfCvgP9+WL9bgfcDf5Tk9e2W0T8EPprkNW3N//VJ3jZ02NuALyMdxtDXCaUF4TsZLGt8E9gL/OPWvInBWvydwKPA94Bfbsc9C/wL4L8xuFL+bjt2pv4T8IEkzyT511P0+a+8dL39h4EvMFhC2cXgiv3n23yeZ/BB7Z+1MS+YYsxfbnPdBdwF/G47z5eoqs3Ah4A/TrKSwQe+pwIPMlh6+j1g+EtjV7T5Si8R/xMVaWaSnAZ8Hbhw4gtaJ6Ik7wSurKp3z/dcdOIx9CWpIy7vSFJHDH1J6oihL0kdMfQlqSMn9DdyFy9eXCtXrpzvaUjSK8q99977RFVN+uXEEzr0V65cyY4dO+Z7GpL0ipJk91RtLu9IUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOnJCfznrlWLlxv8531M4qTz24XfM9xROKv58zp2T4WfTK31J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUkWlDP8mrk9yd5H8n2ZnkP7T6qiRfSzKe5LNJTm3109r+eGtfOTTWda3+cJKLj9tZSZImNZMr/ReAn62qNwLnA5ckuQD4CHBjVb0BeBq4uvW/Gni61W9s/UhyLrAOOA+4BPjtJAvm8FwkSdOYNvRr4Dtt95T2KOBngd9r9c3Au9r25W2f1n5hkrT6bVX1QlU9CowDa+fiJCRJMzOjNf0kC5LcBxwAtgP/F3imqg61LnuBpW17KbAHoLU/C/zocH2SYyRJIzCj0K+qF6vqfGAZg6vzHz9eE0qyIcmOJDsOHjx4vF5Gkro0q7t3quoZ4CvATwOLkkz87p5lwL62vQ9YDtDazwSeHK5Pcszwa9xcVWuqas3Y2KT/mbsk6SjN5O6dsSSL2vYPAT8HPMQg/P9R67Ye+GLb3tr2ae1/XFXV6uva3T2rgNXA3XN0HpKkGZjJb9k8G9jc7rR5FbClqr6U5EHgtiS/DnwduKX1vwX4TJJx4CkGd+xQVTuTbAEeBA4B11TVi3N7OpKkI5k29KvqfuBNk9R3McndN1X1PeAXpxjrBuCG2U9TkjQX/EauJHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdcTQl6SOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI9OGfpLlSb6S5MEkO5P8Sqt/MMm+JPe1x2VDx1yXZDzJw0kuHqpf0mrjSTYen1OSJE1l4Qz6HAJ+tar+IskZwL1Jtre2G6vqN4Y7JzkXWAecB5wD/FGSH2vNnwB+DtgL3JNka1U9OBcnIkma3rShX1X7gf1t+7kkDwFLj3DI5cBtVfUC8GiScWBtaxuvql0ASW5rfQ19SRqRWa3pJ1kJvAn4Witdm+T+JJuSnNVqS4E9Q4ftbbWp6pKkEZlx6Cc5Hfgc8L6q+jZwE/B64HwG/xL46FxMKMmGJDuS7Dh48OBcDClJamYU+klOYRD4v1NVnweoqser6sWq+gHwSf5mCWcfsHzo8GWtNlX9Jarq5qpaU1VrxsbGZns+kqQjmMndOwFuAR6qqo8N1c8e6vYLwANteyuwLslpSVYBq4G7gXuA1UlWJTmVwYe9W+fmNCRJMzGTu3feClwJfCPJfa32fuCKJOcDBTwG/BJAVe1MsoXBB7SHgGuq6kWAJNcCtwMLgE1VtXPOzkSSNK2Z3L1zF5BJmrYd4ZgbgBsmqW870nGSpOPLb+RKUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdcTQl6SOGPqS1BFDX5I6Mm3oJ1me5CtJHkyyM8mvtPprk2xP8kh7PqvVk+TjScaT3J/kzUNjrW/9H0my/vidliRpMjO50j8E/GpVnQtcAFyT5FxgI3BHVa0G7mj7AJcCq9tjA3ATDN4kgOuBtwBrgesn3igkSaMxbehX1f6q+ou2/RzwELAUuBzY3LptBt7Vti8HPl0DXwUWJTkbuBjYXlVPVdXTwHbgkrk8GUnSkc1qTT/JSuBNwNeAJVW1vzV9C1jStpcCe4YO29tqU9UlSSMy49BPcjrwOeB9VfXt4baqKqDmYkJJNiTZkWTHwYMH52JISVIzo9BPcgqDwP+dqvp8Kz/elm1ozwdafR+wfOjwZa02Vf0lqurmqlpTVWvGxsZmcy6SpGnM5O6dALcAD1XVx4aatgITd+CsB744VH9Pu4vnAuDZtgx0O3BRkrPaB7gXtZokaUQWzqDPW4ErgW8kua/V3g98GNiS5GpgN/Du1rYNuAwYB54HrgKoqqeS/BpwT+v3oap6ai5OQpI0M9OGflXdBWSK5gsn6V/ANVOMtQnYNJsJSpLmjt/IlaSOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdWTa0E+yKcmBJA8M1T6YZF+S+9rjsqG265KMJ3k4ycVD9UtabTzJxrk/FUnSdGZypf8p4JJJ6jdW1fntsQ0gybnAOuC8dsxvJ1mQZAHwCeBS4FzgitZXkjRCC6frUFV3Jlk5w/EuB26rqheAR5OMA2tb23hV7QJIclvr++DspyxJOlrHsqZ/bZL72/LPWa22FNgz1Gdvq01Vf5kkG5LsSLLj4MGDxzA9SdLhjjb0bwJeD5wP7Ac+OlcTqqqbq2pNVa0ZGxubq2ElScxgeWcyVfX4xHaSTwJfarv7gOVDXZe1GkeoS5JG5Kiu9JOcPbT7C8DEnT1bgXVJTkuyClgN3A3cA6xOsirJqQw+7N169NOWJB2Naa/0k9wKvB1YnGQvcD3w9iTnAwU8BvwSQFXtTLKFwQe0h4BrqurFNs61wO3AAmBTVe2c65ORJB3ZTO7euWKS8i1H6H8DcMMk9W3AtlnNTpI0p/xGriR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdcTQl6SOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdmTb0k2xKciDJA0O11ybZnuSR9nxWqyfJx5OMJ7k/yZuHjlnf+j+SZP3xOR1J0pHM5Er/U8Alh9U2AndU1WrgjrYPcCmwuj02ADfB4E0CuB54C7AWuH7ijUKSNDrThn5V3Qk8dVj5cmBz294MvGuo/uka+CqwKMnZwMXA9qp6qqqeBrbz8jcSSdJxdrRr+kuqan/b/hawpG0vBfYM9dvbalPVJUkjdMwf5FZVATUHcwEgyYYkO5LsOHjw4FwNK0ni6EP/8bZsQ3s+0Or7gOVD/Za12lT1l6mqm6tqTVWtGRsbO8rpSZImc7ShvxWYuANnPfDFofp72l08FwDPtmWg24GLkpzVPsC9qNUkSSO0cLoOSW4F3g4sTrKXwV04Hwa2JLka2A28u3XfBlwGjAPPA1cBVNVTSX4NuKf1+1BVHf7hsCTpOJs29KvqiimaLpykbwHXTDHOJmDTrGYnSZpTfiNXkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdcTQl6SOGPqS1BFDX5I6YuhLUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUkWMK/SSPJflGkvuS7Gi11ybZnuSR9nxWqyfJx5OMJ7k/yZvn4gQkSTM3F1f6P1NV51fVmra/EbijqlYDd7R9gEuB1e2xAbhpDl5bkjQLx2N553Jgc9veDLxrqP7pGvgqsCjJ2cfh9SVJUzjW0C/gD5Pcm2RDqy2pqv1t+1vAkra9FNgzdOzeVpMkjcjCYzz+71fVviR/C9ie5P8MN1ZVJanZDNjePDYArFix4hinJ0kadkxX+lW1rz0fAL4ArAUen1i2ac8HWvd9wPKhw5e12uFj3lxVa6pqzdjY2LFMT5J0mKMO/SQ/kuSMiW3gIuABYCuwvnVbD3yxbW8F3tPu4rkAeHZoGUiSNALHsryzBPhCkolxfreq/iDJPcCWJFcDu4F3t/7bgMuAceB54KpjeG1J0lE46tCvql3AGyepPwlcOEm9gGuO9vUkScfOb+RKUkcMfUnqiKEvSR0x9CWpI4a+JHXE0Jekjhj6ktQRQ1+SOmLoS1JHDH1J6oihL0kdMfQlqSOGviR1xNCXpI4Y+pLUEUNfkjpi6EtSRwx9SeqIoS9JHTH0Jakjhr4kdcTQl6SOGPqS1BFDX5I6MvLQT3JJkoeTjCfZOOrXl6SejTT0kywAPgFcCpwLXJHk3FHOQZJ6Nuor/bXAeFXtqqrvA7cBl494DpLUrYUjfr2lwJ6h/b3AW4Y7JNkAbGi730ny8Ijm1oPFwBPzPYnp5CPzPQPNkxP+5/MV9LP5uqkaRh3606qqm4Gb53seJ6MkO6pqzXzPQ5qMP5+jMerlnX3A8qH9Za0mSRqBUYf+PcDqJKuSnAqsA7aOeA6S1K2RLu9U1aEk1wK3AwuATVW1c5Rz6JzLZjqR+fM5Aqmq+Z6DJGlE/EauJHXE0Jekjhj6ktSRE+4+fc2dJD/O4BvPS1tpH7C1qh6av1lJmk9e6Z+kkvw7Br/mIsDd7RHgVn/RnU5kSa6a7zmczLx75ySV5C+B86rqrw+rnwrsrKrV8zMz6ciSfLOqVsz3PE5WLu+cvH4AnAPsPqx+dmuT5k2S+6dqApaMci69MfRPXu8D7kjyCH/zS+5WAG8Arp2vSUnNEuBi4OnD6gH+fPTT6Yehf5Kqqj9I8mMMfp318Ae591TVi/M3MwmALwGnV9V9hzck+ZORz6YjrulLUke8e0eSOmLoS1JHDH1J6oihL0kdMfQlqSP/D5YtAzTnzALtAAAAAElFTkSuQmCC\n",
      "text/plain": [
       "<Figure size 432x288 with 1 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "class_count_1, class_count_2 = train['stroke'].value_counts()\n",
    "#seprate 2 classes of Outcome\n",
    "class_1 = train[train['stroke'] == 0]\n",
    "class_2 = train[train['stroke'] == 1]# print the shape of the class\n",
    "#print('class 1:', class_1.shape)\n",
    "#print('class 2:', class_2.shape)\n",
    "class_2_over = class_2.sample(class_count_1, replace=True)\n",
    "test_over = pd.concat([class_2_over, class_1], axis=0)\n",
    "#print(\"total Outcome of 1 and 2:\",test_over['Outcome'].value_counts())# plot the count after under-sampeling\n",
    "test_over['stroke'].value_counts().plot(kind='bar', title='count (Stroke)')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 276,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>hypertension</th>\n",
       "      <th>heart_disease</th>\n",
       "      <th>bmi</th>\n",
       "      <th>work_type_Govt_job</th>\n",
       "      <th>work_type_Never_worked</th>\n",
       "      <th>work_type_Private</th>\n",
       "      <th>work_type_Self-employed</th>\n",
       "      <th>smoking_status_Unknown</th>\n",
       "      <th>smoking_status_formerly smoked</th>\n",
       "      <th>smoking_status_never smoked</th>\n",
       "      <th>...</th>\n",
       "      <th>age_6</th>\n",
       "      <th>age_7</th>\n",
       "      <th>age_8</th>\n",
       "      <th>age_9</th>\n",
       "      <th>age_10</th>\n",
       "      <th>avg_glucose_level_2</th>\n",
       "      <th>avg_glucose_level_3</th>\n",
       "      <th>avg_glucose_level_4</th>\n",
       "      <th>avg_glucose_level_5</th>\n",
       "      <th>avg_glucose_level_6</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>138</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.454757</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4015</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.208046</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>85</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.576518</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>174</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.454757</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4043</th>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>-0.321621</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2825</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.186343</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2046</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.709044</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2112</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-1.570794</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3219</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.322231</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>272</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.878569</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>6332 rows × 32 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      hypertension  heart_disease       bmi  work_type_Govt_job  \\\n",
       "138              0              0 -0.454757                   1   \n",
       "4015             0              0 -0.208046                   0   \n",
       "85               0              0  0.576518                   0   \n",
       "174              0              0 -0.454757                   1   \n",
       "4043             1              1 -0.321621                   0   \n",
       "...            ...            ...       ...                 ...   \n",
       "2825             0              0 -0.186343                   1   \n",
       "2046             0              0 -0.709044                   0   \n",
       "2112             0              0 -1.570794                   0   \n",
       "3219             0              0  0.322231                   1   \n",
       "272              0              0 -0.878569                   1   \n",
       "\n",
       "      work_type_Never_worked  work_type_Private  work_type_Self-employed  \\\n",
       "138                        0                  0                        0   \n",
       "4015                       0                  1                        0   \n",
       "85                         0                  1                        0   \n",
       "174                        0                  0                        0   \n",
       "4043                       0                  0                        1   \n",
       "...                      ...                ...                      ...   \n",
       "2825                       0                  0                        0   \n",
       "2046                       0                  1                        0   \n",
       "2112                       0                  1                        0   \n",
       "3219                       0                  0                        0   \n",
       "272                        0                  0                        0   \n",
       "\n",
       "      smoking_status_Unknown  smoking_status_formerly smoked  \\\n",
       "138                        1                               0   \n",
       "4015                       1                               0   \n",
       "85                         0                               1   \n",
       "174                        0                               1   \n",
       "4043                       0                               0   \n",
       "...                      ...                             ...   \n",
       "2825                       0                               0   \n",
       "2046                       0                               0   \n",
       "2112                       0                               0   \n",
       "3219                       1                               0   \n",
       "272                        0                               1   \n",
       "\n",
       "      smoking_status_never smoked  ...  age_6  age_7  age_8  age_9  age_10  \\\n",
       "138                             0  ...      0      1      0      0       0   \n",
       "4015                            0  ...      0      1      0      0       0   \n",
       "85                              0  ...      0      0      0      1       0   \n",
       "174                             0  ...      1      0      0      0       0   \n",
       "4043                            1  ...      0      0      0      0       1   \n",
       "...                           ...  ...    ...    ...    ...    ...     ...   \n",
       "2825                            1  ...      1      0      0      0       0   \n",
       "2046                            1  ...      0      0      0      0       0   \n",
       "2112                            1  ...      0      0      0      0       0   \n",
       "3219                            0  ...      0      0      0      0       0   \n",
       "272                             0  ...      0      0      0      0       0   \n",
       "\n",
       "      avg_glucose_level_2  avg_glucose_level_3  avg_glucose_level_4  \\\n",
       "138                     0                    0                    1   \n",
       "4015                    0                    0                    0   \n",
       "85                      0                    0                    0   \n",
       "174                     0                    1                    0   \n",
       "4043                    0                    0                    0   \n",
       "...                   ...                  ...                  ...   \n",
       "2825                    1                    0                    0   \n",
       "2046                    1                    0                    0   \n",
       "2112                    1                    0                    0   \n",
       "3219                    0                    0                    0   \n",
       "272                     1                    0                    0   \n",
       "\n",
       "      avg_glucose_level_5  avg_glucose_level_6  \n",
       "138                     0                    0  \n",
       "4015                    0                    0  \n",
       "85                      0                    1  \n",
       "174                     0                    0  \n",
       "4043                    0                    0  \n",
       "...                   ...                  ...  \n",
       "2825                    0                    0  \n",
       "2046                    0                    0  \n",
       "2112                    0                    0  \n",
       "3219                    0                    0  \n",
       "272                     0                    0  \n",
       "\n",
       "[6332 rows x 32 columns]"
      ]
     },
     "execution_count": 276,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "y_train=test_over['stroke']\n",
    "x_train=test_over.drop(columns='stroke')\n",
    "x_train"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 277,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>hypertension</th>\n",
       "      <th>heart_disease</th>\n",
       "      <th>bmi</th>\n",
       "      <th>work_type_Govt_job</th>\n",
       "      <th>work_type_Never_worked</th>\n",
       "      <th>work_type_Private</th>\n",
       "      <th>work_type_Self-employed</th>\n",
       "      <th>smoking_status_Unknown</th>\n",
       "      <th>smoking_status_formerly smoked</th>\n",
       "      <th>smoking_status_never smoked</th>\n",
       "      <th>...</th>\n",
       "      <th>age_6</th>\n",
       "      <th>age_7</th>\n",
       "      <th>age_8</th>\n",
       "      <th>age_9</th>\n",
       "      <th>age_10</th>\n",
       "      <th>avg_glucose_level_2</th>\n",
       "      <th>avg_glucose_level_3</th>\n",
       "      <th>avg_glucose_level_4</th>\n",
       "      <th>avg_glucose_level_5</th>\n",
       "      <th>avg_glucose_level_6</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>3850</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.590645</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2600</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.172216</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2482</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.567774</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3216</th>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0.053817</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1072</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.200470</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>175</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.025563</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3200</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.680790</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>796</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-1.104602</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2636</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>-0.694917</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4164</th>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.082604</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>839 rows × 32 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "      hypertension  heart_disease       bmi  work_type_Govt_job  \\\n",
       "3850             0              0  0.590645                   0   \n",
       "2600             0              0 -0.172216                   0   \n",
       "2482             0              0 -0.567774                   0   \n",
       "3216             0              1  0.053817                   0   \n",
       "1072             0              0 -0.200470                   0   \n",
       "...            ...            ...       ...                 ...   \n",
       "175              0              0  0.025563                   0   \n",
       "3200             0              0 -0.680790                   0   \n",
       "796              0              0 -1.104602                   0   \n",
       "2636             0              0 -0.694917                   0   \n",
       "4164             0              0  0.082604                   0   \n",
       "\n",
       "      work_type_Never_worked  work_type_Private  work_type_Self-employed  \\\n",
       "3850                       0                  1                        0   \n",
       "2600                       0                  1                        0   \n",
       "2482                       0                  1                        0   \n",
       "3216                       0                  1                        0   \n",
       "1072                       0                  1                        0   \n",
       "...                      ...                ...                      ...   \n",
       "175                        0                  1                        0   \n",
       "3200                       0                  1                        0   \n",
       "796                        0                  1                        0   \n",
       "2636                       0                  1                        0   \n",
       "4164                       0                  1                        0   \n",
       "\n",
       "      smoking_status_Unknown  smoking_status_formerly smoked  \\\n",
       "3850                       0                               0   \n",
       "2600                       1                               0   \n",
       "2482                       1                               0   \n",
       "3216                       0                               0   \n",
       "1072                       0                               0   \n",
       "...                      ...                             ...   \n",
       "175                        0                               0   \n",
       "3200                       0                               1   \n",
       "796                        0                               0   \n",
       "2636                       0                               0   \n",
       "4164                       0                               0   \n",
       "\n",
       "      smoking_status_never smoked  ...  age_6  age_7  age_8  age_9  age_10  \\\n",
       "3850                            1  ...      0      0      0      0       0   \n",
       "2600                            0  ...      0      0      0      0       0   \n",
       "2482                            0  ...      1      0      0      0       0   \n",
       "3216                            1  ...      1      0      0      0       0   \n",
       "1072                            1  ...      0      0      0      0       0   \n",
       "...                           ...  ...    ...    ...    ...    ...     ...   \n",
       "175                             1  ...      0      0      1      0       0   \n",
       "3200                            0  ...      0      0      0      1       0   \n",
       "796                             1  ...      0      0      0      0       0   \n",
       "2636                            1  ...      0      1      0      0       0   \n",
       "4164                            0  ...      0      0      0      0       0   \n",
       "\n",
       "      avg_glucose_level_2  avg_glucose_level_3  avg_glucose_level_4  \\\n",
       "3850                    1                    0                    0   \n",
       "2600                    1                    0                    0   \n",
       "2482                    1                    0                    0   \n",
       "3216                    1                    0                    0   \n",
       "1072                    0                    0                    1   \n",
       "...                   ...                  ...                  ...   \n",
       "175                     0                    0                    0   \n",
       "3200                    1                    0                    0   \n",
       "796                     0                    0                    0   \n",
       "2636                    1                    0                    0   \n",
       "4164                    0                    1                    0   \n",
       "\n",
       "      avg_glucose_level_5  avg_glucose_level_6  \n",
       "3850                    0                    0  \n",
       "2600                    0                    0  \n",
       "2482                    0                    0  \n",
       "3216                    0                    0  \n",
       "1072                    0                    0  \n",
       "...                   ...                  ...  \n",
       "175                     1                    0  \n",
       "3200                    0                    0  \n",
       "796                     0                    0  \n",
       "2636                    0                    0  \n",
       "4164                    0                    0  \n",
       "\n",
       "[839 rows x 32 columns]"
      ]
     },
     "execution_count": 277,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "y_test=test['stroke']\n",
    "x_test=test.drop(columns='stroke')\n",
    "x_test"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 278,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "acc for training data: 0.966\n",
      "acc for test data: 0.846\n"
     ]
    }
   ],
   "source": [
    "mlp = MLPClassifier(hidden_layer_sizes=(25,50, 100), max_iter=1000,alpha=1)\n",
    "mlp.fit(x_train, y_train.values.ravel())\n",
    "y_predicted=mlp.predict(x_test)\n",
    "print('acc for training data: {:.3f}'.format(mlp.score(x_train, y_train)))\n",
    "print('acc for test data: {:.3f}'.format(mlp.score(x_test, y_test)))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 279,
   "metadata": {
    "pycharm": {
     "is_executing": false,
     "name": "#%%\n"
    }
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[[692  89]\n",
      " [ 40  18]]\n",
      "              precision    recall  f1-score   support\n",
      "\n",
      "           0       0.95      0.89      0.91       781\n",
      "           1       0.17      0.31      0.22        58\n",
      "\n",
      "    accuracy                           0.85       839\n",
      "   macro avg       0.56      0.60      0.57       839\n",
      "weighted avg       0.89      0.85      0.87       839\n",
      "\n"
     ]
    }
   ],
   "source": [
    "print(confusion_matrix(y_test, y_predicted))\n",
    "print(classification_report(y_test, y_predicted))"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.2"
  },
  "pycharm": {
   "stem_cell": {
    "cell_type": "raw",
    "metadata": {
     "collapsed": false
    },
    "source": []
   }
  }
 },
 "nbformat": 4,
 "nbformat_minor": 1
}