heart_disease.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np \n",
    "import pandas as pd\n",
    "import matplotlib.pyplot as plt\n",
    "import seaborn as sns\n",
    "from sklearn.model_selection import train_test_split\n",
    "from sklearn.linear_model import LogisticRegression\n",
    "from sklearn.metrics import accuracy_score, confusion_matrix"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "dataset_dir = \"./datasets/\"\n",
    "heart_disease = pd.read_csv(dataset_dir + \"heart_disease_data.csv\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "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>sex</th>\n",
       "      <th>cp</th>\n",
       "      <th>trestbps</th>\n",
       "      <th>chol</th>\n",
       "      <th>fbs</th>\n",
       "      <th>restecg</th>\n",
       "      <th>thalach</th>\n",
       "      <th>exang</th>\n",
       "      <th>oldpeak</th>\n",
       "      <th>slope</th>\n",
       "      <th>ca</th>\n",
       "      <th>thal</th>\n",
       "      <th>target</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>63</td>\n",
       "      <td>1</td>\n",
       "      <td>3</td>\n",
       "      <td>145</td>\n",
       "      <td>233</td>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>150</td>\n",
       "      <td>0</td>\n",
       "      <td>2.3</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>37</td>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>130</td>\n",
       "      <td>250</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>187</td>\n",
       "      <td>0</td>\n",
       "      <td>3.5</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>41</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>130</td>\n",
       "      <td>204</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>172</td>\n",
       "      <td>0</td>\n",
       "      <td>1.4</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>56</td>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>120</td>\n",
       "      <td>236</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>178</td>\n",
       "      <td>0</td>\n",
       "      <td>0.8</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>57</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>120</td>\n",
       "      <td>354</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>163</td>\n",
       "      <td>1</td>\n",
       "      <td>0.6</td>\n",
       "      <td>2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   age  sex  cp  trestbps  chol  fbs  restecg  thalach  exang  oldpeak  slope  \\\n",
       "0   63    1   3       145   233    1        0      150      0      2.3      0   \n",
       "1   37    1   2       130   250    0        1      187      0      3.5      0   \n",
       "2   41    0   1       130   204    0        0      172      0      1.4      2   \n",
       "3   56    1   1       120   236    0        1      178      0      0.8      2   \n",
       "4   57    0   0       120   354    0        1      163      1      0.6      2   \n",
       "\n",
       "   ca  thal  target  \n",
       "0   0     1       1  \n",
       "1   0     2       1  \n",
       "2   0     2       1  \n",
       "3   0     2       1  \n",
       "4   0     2       1  "
      ]
     },
     "execution_count": 14,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "heart_disease.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "<class 'pandas.core.frame.DataFrame'>\n",
      "RangeIndex: 303 entries, 0 to 302\n",
      "Data columns (total 14 columns):\n",
      " #   Column    Non-Null Count  Dtype  \n",
      "---  ------    --------------  -----  \n",
      " 0   age       303 non-null    int64  \n",
      " 1   sex       303 non-null    int64  \n",
      " 2   cp        303 non-null    int64  \n",
      " 3   trestbps  303 non-null    int64  \n",
      " 4   chol      303 non-null    int64  \n",
      " 5   fbs       303 non-null    int64  \n",
      " 6   restecg   303 non-null    int64  \n",
      " 7   thalach   303 non-null    int64  \n",
      " 8   exang     303 non-null    int64  \n",
      " 9   oldpeak   303 non-null    float64\n",
      " 10  slope     303 non-null    int64  \n",
      " 11  ca        303 non-null    int64  \n",
      " 12  thal      303 non-null    int64  \n",
      " 13  target    303 non-null    int64  \n",
      "dtypes: float64(1), int64(13)\n",
      "memory usage: 33.3 KB\n"
     ]
    }
   ],
   "source": [
    "heart_disease.info()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "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>sex</th>\n",
       "      <th>cp</th>\n",
       "      <th>trestbps</th>\n",
       "      <th>chol</th>\n",
       "      <th>fbs</th>\n",
       "      <th>restecg</th>\n",
       "      <th>thalach</th>\n",
       "      <th>exang</th>\n",
       "      <th>oldpeak</th>\n",
       "      <th>slope</th>\n",
       "      <th>ca</th>\n",
       "      <th>thal</th>\n",
       "      <th>target</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>count</th>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "      <td>303.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>mean</th>\n",
       "      <td>54.366337</td>\n",
       "      <td>0.683168</td>\n",
       "      <td>0.966997</td>\n",
       "      <td>131.623762</td>\n",
       "      <td>246.264026</td>\n",
       "      <td>0.148515</td>\n",
       "      <td>0.528053</td>\n",
       "      <td>149.646865</td>\n",
       "      <td>0.326733</td>\n",
       "      <td>1.039604</td>\n",
       "      <td>1.399340</td>\n",
       "      <td>0.729373</td>\n",
       "      <td>2.313531</td>\n",
       "      <td>0.544554</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>std</th>\n",
       "      <td>9.082101</td>\n",
       "      <td>0.466011</td>\n",
       "      <td>1.032052</td>\n",
       "      <td>17.538143</td>\n",
       "      <td>51.830751</td>\n",
       "      <td>0.356198</td>\n",
       "      <td>0.525860</td>\n",
       "      <td>22.905161</td>\n",
       "      <td>0.469794</td>\n",
       "      <td>1.161075</td>\n",
       "      <td>0.616226</td>\n",
       "      <td>1.022606</td>\n",
       "      <td>0.612277</td>\n",
       "      <td>0.498835</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>min</th>\n",
       "      <td>29.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>94.000000</td>\n",
       "      <td>126.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>71.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25%</th>\n",
       "      <td>47.500000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>120.000000</td>\n",
       "      <td>211.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>133.500000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>0.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>50%</th>\n",
       "      <td>55.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>130.000000</td>\n",
       "      <td>240.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>153.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>0.800000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>75%</th>\n",
       "      <td>61.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>140.000000</td>\n",
       "      <td>274.500000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>166.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.600000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>3.000000</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>max</th>\n",
       "      <td>77.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>3.000000</td>\n",
       "      <td>200.000000</td>\n",
       "      <td>564.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>202.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>6.200000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>4.000000</td>\n",
       "      <td>3.000000</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "              age         sex          cp    trestbps        chol         fbs  \\\n",
       "count  303.000000  303.000000  303.000000  303.000000  303.000000  303.000000   \n",
       "mean    54.366337    0.683168    0.966997  131.623762  246.264026    0.148515   \n",
       "std      9.082101    0.466011    1.032052   17.538143   51.830751    0.356198   \n",
       "min     29.000000    0.000000    0.000000   94.000000  126.000000    0.000000   \n",
       "25%     47.500000    0.000000    0.000000  120.000000  211.000000    0.000000   \n",
       "50%     55.000000    1.000000    1.000000  130.000000  240.000000    0.000000   \n",
       "75%     61.000000    1.000000    2.000000  140.000000  274.500000    0.000000   \n",
       "max     77.000000    1.000000    3.000000  200.000000  564.000000    1.000000   \n",
       "\n",
       "          restecg     thalach       exang     oldpeak       slope          ca  \\\n",
       "count  303.000000  303.000000  303.000000  303.000000  303.000000  303.000000   \n",
       "mean     0.528053  149.646865    0.326733    1.039604    1.399340    0.729373   \n",
       "std      0.525860   22.905161    0.469794    1.161075    0.616226    1.022606   \n",
       "min      0.000000   71.000000    0.000000    0.000000    0.000000    0.000000   \n",
       "25%      0.000000  133.500000    0.000000    0.000000    1.000000    0.000000   \n",
       "50%      1.000000  153.000000    0.000000    0.800000    1.000000    0.000000   \n",
       "75%      1.000000  166.000000    1.000000    1.600000    2.000000    1.000000   \n",
       "max      2.000000  202.000000    1.000000    6.200000    2.000000    4.000000   \n",
       "\n",
       "             thal      target  \n",
       "count  303.000000  303.000000  \n",
       "mean     2.313531    0.544554  \n",
       "std      0.612277    0.498835  \n",
       "min      0.000000    0.000000  \n",
       "25%      2.000000    0.000000  \n",
       "50%      2.000000    1.000000  \n",
       "75%      3.000000    1.000000  \n",
       "max      3.000000    1.000000  "
      ]
     },
     "execution_count": 16,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "heart_disease.describe()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "age         0\n",
       "sex         0\n",
       "cp          0\n",
       "trestbps    0\n",
       "chol        0\n",
       "fbs         0\n",
       "restecg     0\n",
       "thalach     0\n",
       "exang       0\n",
       "oldpeak     0\n",
       "slope       0\n",
       "ca          0\n",
       "thal        0\n",
       "target      0\n",
       "dtype: int64"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "heart_disease.isnull().sum()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "1    165\n",
       "0    138\n",
       "Name: target, dtype: int64"
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "heart_disease['target'].value_counts()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [],
   "source": [
    "X = heart_disease.drop(columns='target', axis=1)\n",
    "Y = heart_disease['target']\n",
    "X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.2, stratify=Y, random_state=2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/opt/homebrew/Caskroom/miniconda/base/envs/ai/lib/python3.9/site-packages/sklearn/linear_model/_logistic.py:458: ConvergenceWarning: lbfgs failed to converge (status=1):\n",
      "STOP: TOTAL NO. of ITERATIONS REACHED LIMIT.\n",
      "\n",
      "Increase the number of iterations (max_iter) or scale the data as shown in:\n",
      "    https://scikit-learn.org/stable/modules/preprocessing.html\n",
      "Please also refer to the documentation for alternative solver options:\n",
      "    https://scikit-learn.org/stable/modules/linear_model.html#logistic-regression\n",
      "  n_iter_i = _check_optimize_result(\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<style>#sk-container-id-2 {color: black;background-color: white;}#sk-container-id-2 pre{padding: 0;}#sk-container-id-2 div.sk-toggleable {background-color: white;}#sk-container-id-2 label.sk-toggleable__label {cursor: pointer;display: block;width: 100%;margin-bottom: 0;padding: 0.3em;box-sizing: border-box;text-align: center;}#sk-container-id-2 label.sk-toggleable__label-arrow:before {content: \"▸\";float: left;margin-right: 0.25em;color: #696969;}#sk-container-id-2 label.sk-toggleable__label-arrow:hover:before {color: black;}#sk-container-id-2 div.sk-estimator:hover label.sk-toggleable__label-arrow:before {color: black;}#sk-container-id-2 div.sk-toggleable__content {max-height: 0;max-width: 0;overflow: hidden;text-align: left;background-color: #f0f8ff;}#sk-container-id-2 div.sk-toggleable__content pre {margin: 0.2em;color: black;border-radius: 0.25em;background-color: #f0f8ff;}#sk-container-id-2 input.sk-toggleable__control:checked~div.sk-toggleable__content {max-height: 200px;max-width: 100%;overflow: auto;}#sk-container-id-2 input.sk-toggleable__control:checked~label.sk-toggleable__label-arrow:before {content: \"▾\";}#sk-container-id-2 div.sk-estimator input.sk-toggleable__control:checked~label.sk-toggleable__label {background-color: #d4ebff;}#sk-container-id-2 div.sk-label input.sk-toggleable__control:checked~label.sk-toggleable__label {background-color: #d4ebff;}#sk-container-id-2 input.sk-hidden--visually {border: 0;clip: rect(1px 1px 1px 1px);clip: rect(1px, 1px, 1px, 1px);height: 1px;margin: -1px;overflow: hidden;padding: 0;position: absolute;width: 1px;}#sk-container-id-2 div.sk-estimator {font-family: monospace;background-color: #f0f8ff;border: 1px dotted black;border-radius: 0.25em;box-sizing: border-box;margin-bottom: 0.5em;}#sk-container-id-2 div.sk-estimator:hover {background-color: #d4ebff;}#sk-container-id-2 div.sk-parallel-item::after {content: \"\";width: 100%;border-bottom: 1px solid gray;flex-grow: 1;}#sk-container-id-2 div.sk-label:hover label.sk-toggleable__label {background-color: #d4ebff;}#sk-container-id-2 div.sk-serial::before {content: \"\";position: absolute;border-left: 1px solid gray;box-sizing: border-box;top: 0;bottom: 0;left: 50%;z-index: 0;}#sk-container-id-2 div.sk-serial {display: flex;flex-direction: column;align-items: center;background-color: white;padding-right: 0.2em;padding-left: 0.2em;position: relative;}#sk-container-id-2 div.sk-item {position: relative;z-index: 1;}#sk-container-id-2 div.sk-parallel {display: flex;align-items: stretch;justify-content: center;background-color: white;position: relative;}#sk-container-id-2 div.sk-item::before, #sk-container-id-2 div.sk-parallel-item::before {content: \"\";position: absolute;border-left: 1px solid gray;box-sizing: border-box;top: 0;bottom: 0;left: 50%;z-index: -1;}#sk-container-id-2 div.sk-parallel-item {display: flex;flex-direction: column;z-index: 1;position: relative;background-color: white;}#sk-container-id-2 div.sk-parallel-item:first-child::after {align-self: flex-end;width: 50%;}#sk-container-id-2 div.sk-parallel-item:last-child::after {align-self: flex-start;width: 50%;}#sk-container-id-2 div.sk-parallel-item:only-child::after {width: 0;}#sk-container-id-2 div.sk-dashed-wrapped {border: 1px dashed gray;margin: 0 0.4em 0.5em 0.4em;box-sizing: border-box;padding-bottom: 0.4em;background-color: white;}#sk-container-id-2 div.sk-label label {font-family: monospace;font-weight: bold;display: inline-block;line-height: 1.2em;}#sk-container-id-2 div.sk-label-container {text-align: center;}#sk-container-id-2 div.sk-container {/* jupyter's `normalize.less` sets `[hidden] { display: none; }` but bootstrap.min.css set `[hidden] { display: none !important; }` so we also need the `!important` here to be able to override the default hidden behavior on the sphinx rendered scikit-learn.org. See: https://github.com/scikit-learn/scikit-learn/issues/21755 */display: inline-block !important;position: relative;}#sk-container-id-2 div.sk-text-repr-fallback {display: none;}</style><div id=\"sk-container-id-2\" class=\"sk-top-container\"><div class=\"sk-text-repr-fallback\"><pre>LogisticRegression()</pre><b>In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook. <br />On GitHub, the HTML representation is unable to render, please try loading this page with nbviewer.org.</b></div><div class=\"sk-container\" hidden><div class=\"sk-item\"><div class=\"sk-estimator sk-toggleable\"><input class=\"sk-toggleable__control sk-hidden--visually\" id=\"sk-estimator-id-2\" type=\"checkbox\" checked><label for=\"sk-estimator-id-2\" class=\"sk-toggleable__label sk-toggleable__label-arrow\">LogisticRegression</label><div class=\"sk-toggleable__content\"><pre>LogisticRegression()</pre></div></div></div></div></div>"
      ],
      "text/plain": [
       "LogisticRegression()"
      ]
     },
     "execution_count": 20,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "model = LogisticRegression()\n",
    "model.fit(X_train, Y_train)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Accuracy on test data :  0.819672131147541\n"
     ]
    }
   ],
   "source": [
    "test_prediction = model.predict(X_test)\n",
    "test_accuracy = accuracy_score(test_prediction, Y_test)\n",
    "print('Accuracy on test data : ', test_accuracy)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAWQAAAD4CAYAAADbyJysAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjcuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/bCgiHAAAACXBIWXMAAAsTAAALEwEAmpwYAAAWLklEQVR4nO3dfbxVVZ3H8c83QEcBDVMJEVGTKNLCQjS1EVMIb5Y9jSM1hkVz1UnLGStJe7aZnClrKn2lVyFrUpwaQ83wgSwjX6WBhgoBQobJBSHFFMVXcu/5zR9nQ6frebqXc+/ZZ/V981qve/baD+un4I/l76y9tyICMzNrvpc0OwAzMytyQjYzywknZDOznHBCNjPLCSdkM7OcGNzfAzx/7ae9jMNeZPgH5jY7BMuhrhc6tbPX2PbEI3XnnCF7H7zT4zVSvydkM7MBVehudgR95oRsZmmJQrMj6DMnZDNLS8EJ2cwsF8IzZDOznOjuanYEfeaEbGZp8Zd6ZmY54ZKFmVlO+Es9M7N88Jd6ZmZ54RmymVlOdG9rdgR95oRsZmlxycLMLCdcsjAzywnPkM3McsIzZDOzfIhC636p5zeGmFlaCoX6WxWSxkj6maTfSlou6aNZ/+ckdUpamrW2CudPl7RK0hpJs+sJ3TNkM0tL42rIXcD5EXG/pOHAfZIWZvu+FhFfqXSipEHA5cBUYB2wWNLNEfHbagM6IZtZWhr0cKGI2ABsyD5vkbQCGF3n6ZOBNRHxCICk64FTgKoJ2SULM0tLFOpuktolLSlp7eUuKelA4HDg3qzrHEkPSporaUSZU0YDj5Vsr6OOZO6EbGZp6UUNOSI6ImJSSevoeTlJw4AbgPMi4hngW8ArgIkUZ9CXNip0lyzMLC0NfEC9pCEUk/G1EfFDgIjYWLL/KuCWMqd2AmNKtvfP+qryDNnM0tK4VRYC5gArIuKrJf2jSg57J7CszOmLgXGSDpK0C3AacHOt0D1DNrOkRDTsjSHHAKcDD0lamvVdCMyQNBEIYC1wJoCk/YCrI6ItIroknQPcDgwC5kbE8loDOiGbWVoadKdeRNwNqMyuBRWOXw+0lWwvqHRsJU7IZpYWP8vCzCwn/CwLM7OcaOAqi4HmhGxmaXHJwswsJ1yyMDPLCSdkM7OccMnCzCwn/KWemVlOuGRhZpYTLlmYmeWEZ8hmZjnhhGxmlhMRzY6gz5yQzSwtXV5lYWaWD/5Sz8wsJ1q4huxXOJlZWiLqb1VIGiPpZ5J+K2m5pI9m/V+WtDJ76/R8SS+tcP5aSQ9JWippST2hOyGbWVoa9E49oAs4PyImAEcBH5Y0AVgIHBoRrwUeBj5Z5RrHR8TEiJhUT+guWZhZWhr3CqcNwIbs8xZJK4DREXFHyWH3AO9pyIB4hmxmiYnu7rqbpHZJS0pae7lrSjoQOBy4t8euDwK3VgoFuEPSfZWu25NnyGaWll7MkCOiA+iodoykYcANwHkR8UxJ/0UUyxrXVjj12IjolLQvsFDSyohYVG0sz5DNLC1RqL/VIGkIxWR8bUT8sKT/DOBk4H0R5b8djIjO7OcmYD4wudZ4TshmlpZC1N+qkCRgDrAiIr5a0j8d+ATw9ojYWuHcoZKGb/8MTAOW1QrdJQszS0vj1iEfA5wOPCRpadZ3IfANYFeKZQiAeyLiLEn7AVdHRBswEpif7R8MXBcRt9Ua0AnZzNLS3d2Qy0TE3YDK7FpQ4fj1QFv2+RHgdb0d0wm5nzz+9FY+deO9bH7uzyB49+sP5n1HvpLLf/YQd61ajyT2GrorXzhlMvsO363Z4VqTrHn4HrY8+yzd3QW6uro46o1tzQ6p9bXwnXpOyP1k0EvE+dMm8upRI3juz9uYcdVCjjp4JDOPfhUfPv4wAK6792E6Fi3nU2+ta824JerEqf/Ak08+1eww0lGjNpxnTsj9ZJ/hu7FPNvMduusQDt57DzY98zyv2GfPHcc8v60blf0/IjPrs5QfLiTpVcApwOisqxO4OSJW9GdgKen803OsfPxPHLb/ywD45k8f4pYH1zJs1yFc9f4pzQ3OmioiuHXBPCKCq676HlfPqbSk1erWwjPkqsveJF0AXE+xsP3rrAmYJ2l2lfN23P0y56f3NzLelrP1hW187Ae/5ONvmciwXYcAcO6bD+P2895G22FjuX7xmiZHaM103PHvZPKR0zn5bf/E2WefwZuOPbLZIbW8KBTqbnlTax3yLOCIiLgkIr6XtUsoLnCeVemkiOiIiEkRMWnWm1/fyHhbyrbuAud//5e0HXoAJ7x6/xftbzvsAO5csa4JkVlerF//OAB//OOT3HTTrRxxxMTmBpSC7u76W87USsgFYL8y/aOyfVZBRPD5Hy3moH324PQ3jt/R/+iTW3Z8vmvVeg7ae49mhGc5sPvuuzFs2NAdn6eeeBzLl69qclQJaNCNIc1Qq4Z8HnCnpNXAY1nfAcAhwDn9GFfLW/rYE9zy4KOM23dPTr2y+HCoc998GDf+5hHWPrmFl0iM2nN3LnrrG5ocqTXLyJH78H8/mAPA4MGDuP76G7n9jruaG1QKcliKqFfVhBwRt0l6JcUSRemXeosjIn/z/Rw5/IB9WPqZU1/U/6Zxo5oQjeXR73//B94waWqzw0hPDme+9aq5yiIiChSf+Wlmln8pL3szM2spKc+QzcxaSXS1bjXVCdnM0uIZsplZTriGbGaWEy08Q/YbQ8wsKVGIuls1ksZI+pmk30paLumjWf9ekhZKWp39HFHh/JnZMaslzawndidkM0tLV3f9rcaVgPMjYgJwFPBhSROA2cCdETEOuDPb/iuS9gI+CxxJ8T6Oz1ZK3KWckM0sLQ26dToiNkTE/dnnLcAKijfInQJ8JzvsO8A7ypz+FmBhRGyOiKeAhcD0WqG7hmxmaemHGrKkA4HDgXuBkRGxIdv1OMX35/U0mr88bgJgHX+527kiz5DNLCkRUXcrfVRw1tp7Xk/SMOAG4LyIeKbHWAE07G8Az5DNLC29mCFHRAfQUWm/pCEUk/G1EfHDrHujpFERsUHSKGBTmVM7gSkl2/sDd9WKxzNkM0tLg2rIkgTMAVZExFdLdt0MbF81MRO4qczptwPTJI3IvsyblvVV5RmymSUluhp2Y8gxwOnAQ5KWZn0XApcA35c0C3gUOBVA0iTgrIj4UERslnQxsDg77wsRsbnWgE7IZpaWBuXjiLgbKr6F+IQyxy8BPlSyPReY25sxnZDNLCm1bvjIMydkM0uLE7KZWU607rOFnJDNLC0uWZiZ5UR0OSGbmeWDSxZmZvnQws+nd0I2s8Q4IZuZ5YNnyGZmORFdzY6g75yQzSwpniGbmeWEE7KZWV5EpecB5Z8TspklxTNkM7OciIJnyGZmuVDodkI2M8uFRpYsJM0FTgY2RcShWd//AuOzQ14K/CkiJpY5dy2wBegGuiJiUq3xnJDNLCkNLllcA1wGfHfH9SP+cftnSZcCT1c5//iIeKLewZyQzSwp0cCHvUXEIkkHltuXvQT1VODNjRrPb502s6REQXU3Se2SlpS09l4M9SZgY0SsrhQKcIek++q9rmfIZpaU3nypFxEdQEcfh5oBzKuy/9iI6JS0L7BQ0sqIWFTtgk7IZpaUgVj2Jmkw8C7gDRXjiOjMfm6SNB+YDFRNyC5ZmFlSIlR32wknAisjYl25nZKGShq+/TMwDVhW66JOyGaWlCjU32qRNA/4FTBe0jpJs7Jdp9GjXCFpP0kLss2RwN2SHgB+Dfw4Im6rNZ5LFmaWlEIDn2URETMq9J9Rpm890JZ9fgR4XW/Hc0I2s6TsZCmiqZyQzSwpvnXazCwn/HAhM7OcaGQNeaA5IZtZUlxDNjPLiUY+y2KgOSGbWVJcsjAzy4mCv9QzM8sHz5CrGH3W9f09hLWg59f/otkhWKL8pZ6ZWU54hmxmlhMtvMjCCdnM0tJdaN2HWDohm1lSGvjS6QHnhGxmSQlcQzYzy4VCCxeRW7fYYmZWRgHV3WqRNFfSJknLSvo+J6lT0tKstVU4d7qkVZLWSJpdT+xOyGaWlEB1tzpcA0wv0/+1iJiYtQU9d0oaBFwOnARMAGZImlBrMCdkM0tKN6q71RIRi4DNfQhjMrAmIh6JiBeA64FTap3khGxmSSn0oklql7SkpLXXOcw5kh7MShojyuwfDTxWsr0u66vKCdnMktKbhBwRHRExqaR11DHEt4BXABOBDcCljYrdqyzMLCn9vewtIjZu/yzpKuCWMod1AmNKtvfP+qryDNnMklJQ/a0vJI0q2XwnsKzMYYuBcZIOkrQLcBpwc61re4ZsZkmpZzlbvSTNA6YAe0taB3wWmCJpIsXHZqwFzsyO3Q+4OiLaIqJL0jnA7cAgYG5ELK81nhOymSWlu4HXiogZZbrnVDh2PdBWsr0AeNGSuGqckM0sKQX51mkzs1xo4TunnZDNLC1+2puZWU608DtOnZDNLC313BKdV07IZpYUz5DNzHLCNWQzs5zwKgszs5xwycLMLCdcsjAzy4luz5DNzPLBM2Qzs5xwQjYzywmvsjAzywmvsjAzy4lWLln4FU5mlpTuXrRasrdKb5K0rKTvy5JWZm+dni/ppRXOXSvpIUlLJS2pJ3YnZDNLSoPfqXcNML1H30Lg0Ih4LfAw8Mkq5x8fERMjYlI9gzkhm1lSCr1otUTEImBzj747IqIr27yH4hulG8IJ2cySEr1oktolLSlp7b0c7oPArVVCuUPSffVe11/qmVlSCr1Y+BYRHUBHX8aRdBHQBVxb4ZBjI6JT0r7AQkkrsxl3RZ4hm1lSGvmlXiWSzgBOBt4XEWX/BoiIzuznJmA+MLnWdZ2QzSwpjawhlyNpOvAJ4O0RsbXCMUMlDd/+GZgGLCt3bCknZDNLSiNXWUiaB/wKGC9pnaRZwGXAcIpliKWSrsiO3U/SguzUkcDdkh4Afg38OCJuqzWea8hmlpTe1JBriYgZZbrnVDh2PdCWfX4EeF1vx3NCNrOk+FkWZmY50cq3Tjshm1lSult4juyEbGZJ8QzZzCwnGvml3kBzQjazpLRuOnZCNrPEuGRhZpYT/lLPzCwnXEO2mvbYczjfuOw/eNWEcRBw7r/MZvGvlzY7LBtAGzb+kQsv/gpPPvUUQrznlJM4/dR3cP6nv8TaP6wDYMuzzzJ82DBu+M7lTY62dbVuOnZCHjBf+q9PcedPFnHG6ecyZMgQdtv975odkg2wwYMG8fFz/5kJ4w/huee2cuqsj3D0EYdz6cV/eeHEl795FcOG7t7EKFtfK8+Q/XChATB8j2EcffQR/M93fgDAtm3beObpLU2OygbaPnvvxYTxhwAwdOjuHDx2DBv/+OSO/RHBbT9dRNvUKU2KMA39/bS3/uSEPADGjh3DE09s5rIr/pO77r6Jr1/27+y++27NDsuaqHPDRlas/h2vfc34HX33PbCMl40Ywdgxo5sYWeuLXvzKmz4nZEkfqLJvx2tR/rzt6b4OkYzBgwfxuomv4dtXX8eUY09h63PPc96/ndnssKxJtm59nn+96Itc8JEzGTZ06I7+BQvvom3qcU2MLA3dRN0tb3Zmhvz5SjsioiMiJkXEpF2H7LkTQ6RhfefjrO98nPuWPADATTfdxmsnvqbJUVkzbOvq4ryLvshbpx3P1CnH7Ojv6urmJz//JdNP+PsmRpeGZEsWkh6s0B6i+ABmq8OmTU/Q2bmBQ8YdBMBxx72RVSvXNDkqG2gRwWe+9N8cPHYMM09711/tu2fJbzh47P68fN99mhRdOgoRdbdaJM2VtEnSspK+vSQtlLQ6+zmiwrkzs2NWS5pZT+y1VlmMBN4CPNVzLOCX9QxgRRd87GKuvPpSdtllCGvXPsY5Z89udkg2wH7z4HJ+dNudjHvFgbx75ocB+OiZM/n7oydz609+zkknTmlugIlocCHiGopvCPluSd9s4M6IuETS7Gz7gtKTJO0FfBaYlIV0n6SbI6JnLv0rqvB+vu0XnQN8OyLuLrPvuoh4b61/mr2Gj8tfocaabuPvb292CJZDQ/Y+uI4XK1X33rHvrDvnXPfo/JrjSToQuCUiDs22VwFTImKDpFHAXRExvsc5M7Jjzsy2r8yOm1dtrKoz5IiYVWVfzWRsZjbQerN6QlI70F7S1RERHTVOGxkRG7LPj1O+fDsaeKxke13WV5VvDDGzpHT1IiFnybdWAq52fkhqWBXA65DNLCkDsA55Y1aqIPu5qcwxncCYku39s76qnJDNLCkDsOztZmD7qomZwE1ljrkdmCZpRLYKY1rWV5UTspklJSLqbrVImgf8ChgvaZ2kWcAlwFRJq4ETs20kTZJ0dRbDZuBiYHHWvpD1VeUaspklpZEPF4qIGRV2nVDm2CXAh0q25wJzezOeE7KZJSWPt0TXywnZzJLSyo/fdEI2s6TUUxvOKydkM0tKHh8aVC8nZDNLSh6fc1wvJ2QzS4pryGZmOdEdrVu0cEI2s6S4ZGFmlhP1PHg+r5yQzSwprZuOnZDNLDH+Us/MLCeckM3McsKrLMzMcsKrLMzMcsLPsjAzy4lWriH7jSFmlpRGvTFE0nhJS0vaM5LO63HMFElPlxzzmZ2J3TNkM0tKd4Oe9xYRq4CJAJIGUXxJ6fwyh/4iIk5uxJhOyGaWlH66U+8E4HcR8Wh/XHw7lyzMLCnRi1+S2iUtKWntFS57GjCvwr43SnpA0q2SXrMzsXuGbGZJ6c0MOSI6gI5qx0jaBXg78Mkyu+8HxkbEs5LagBuBcXUH0INnyGaWlN7MkOt0EnB/RGx80VgRz0TEs9nnBcAQSXv3NXbPkM0sKf1QQ55BhXKFpJcDGyMiJE2mOMl9sq8DOSGbWVIaeeu0pKHAVODMkr6zACLiCuA9wNmSuoDngdNiJ+5McUI2s6Q08tbpiHgOeFmPvitKPl8GXNao8ZyQzSwp4YcLmZnlQyvfOu2EbGZJ8cOFzMxywjNkM7Oc6C64hmxmlgt+QL2ZWU64hmxmlhOuIZuZ5YRnyGZmOeEv9czMcsIlCzOznHDJwswsJ/rpFU4DwgnZzJLidchmZjnhGbKZWU4U/PhNM7N8aOSXepLWAluAbqArIib12C/g60AbsBU4IyLu7+t4TshmlpR+WGVxfEQ8UWHfSRTfMj0OOBL4VvazT/zWaTNLSvSiNcApwHej6B7gpZJG9fVi/T5D3rxltfp7jFYhqT0iOpodh+WL/1w0VtcLnXXnHEntQHtJV0eP34sA7pAUwJVlfp9GA4+VbK/L+jb0LuoilywGVjvg//CsJ/+5aJIswVb7d39sRHRK2hdYKGllRCzqr3hcsjAzqyAiOrOfm4D5wOQeh3QCY0q298/6+sQJ2cysDElDJQ3f/hmYBizrcdjNwPtVdBTwdET0qVwBLlkMNP9vqZXjPxf5NBKYX1zZxmDguoi4TdJZABFxBbCA4pK3NRSXvX1gZwZUKz+Iw8wsJS5ZmJnlhBOymVlOOCEPEEnTJa2StEbS7GbHY80naa6kTZJ6flFkf6OckAeApEHA5RRvs5wAzJA0oblRWQ5cA0xvdhCWH07IA2MysCYiHomIF4DrKd5yaX/DshsMNjc7DssPJ+SBUen2SjOzHZyQzcxywgl5YDT09kozS5MT8sBYDIyTdJCkXYDTKN5yaWa2gxPyAIiILuAc4HZgBfD9iFje3Kis2STNA34FjJe0TtKsZsdkzeVbp83McsIzZDOznHBCNjPLCSdkM7OccEI2M8sJJ2Qzs5xwQjYzywknZDOznPh/sLRxZuhN9oUAAAAASUVORK5CYII=",
      "text/plain": [
       "<Figure size 432x288 with 2 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "# Confusion Matrix\n",
    "cm = confusion_matrix(Y_test, test_prediction)\n",
    "sns.heatmap(cm, annot=True)\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "ai",
   "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.9.16"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}