{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "# Logistic regression with scikit-learn: heart disease data set" ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [], "source": [ "import numpy as np\n", "import pandas as pd\n", "import matplotlib.pyplot as plt" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "### Read data " ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", " | age | \n", "sex | \n", "cp | \n", "trestbps | \n", "chol | \n", "fbs | \n", "restecg | \n", "thalach | \n", "exang | \n", "oldpeak | \n", "slope | \n", "ca | \n", "thal | \n", "target | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | \n", "63 | \n", "1 | \n", "3 | \n", "145 | \n", "233 | \n", "1 | \n", "0 | \n", "150 | \n", "0 | \n", "2.3 | \n", "0 | \n", "0 | \n", "1 | \n", "1 | \n", "
1 | \n", "37 | \n", "1 | \n", "2 | \n", "130 | \n", "250 | \n", "0 | \n", "1 | \n", "187 | \n", "0 | \n", "3.5 | \n", "0 | \n", "0 | \n", "2 | \n", "1 | \n", "
2 | \n", "41 | \n", "0 | \n", "1 | \n", "130 | \n", "204 | \n", "0 | \n", "0 | \n", "172 | \n", "0 | \n", "1.4 | \n", "2 | \n", "0 | \n", "2 | \n", "1 | \n", "
3 | \n", "56 | \n", "1 | \n", "1 | \n", "120 | \n", "236 | \n", "0 | \n", "1 | \n", "178 | \n", "0 | \n", "0.8 | \n", "2 | \n", "0 | \n", "2 | \n", "1 | \n", "
4 | \n", "57 | \n", "0 | \n", "0 | \n", "120 | \n", "354 | \n", "0 | \n", "1 | \n", "163 | \n", "1 | \n", "0.6 | \n", "2 | \n", "0 | \n", "2 | \n", "1 | \n", "
... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "... | \n", "
298 | \n", "57 | \n", "0 | \n", "0 | \n", "140 | \n", "241 | \n", "0 | \n", "1 | \n", "123 | \n", "1 | \n", "0.2 | \n", "1 | \n", "0 | \n", "3 | \n", "0 | \n", "
299 | \n", "45 | \n", "1 | \n", "3 | \n", "110 | \n", "264 | \n", "0 | \n", "1 | \n", "132 | \n", "0 | \n", "1.2 | \n", "1 | \n", "0 | \n", "3 | \n", "0 | \n", "
300 | \n", "68 | \n", "1 | \n", "0 | \n", "144 | \n", "193 | \n", "1 | \n", "1 | \n", "141 | \n", "0 | \n", "3.4 | \n", "1 | \n", "2 | \n", "3 | \n", "0 | \n", "
301 | \n", "57 | \n", "1 | \n", "0 | \n", "130 | \n", "131 | \n", "0 | \n", "1 | \n", "115 | \n", "1 | \n", "1.2 | \n", "1 | \n", "1 | \n", "3 | \n", "0 | \n", "
302 | \n", "57 | \n", "0 | \n", "1 | \n", "130 | \n", "236 | \n", "0 | \n", "0 | \n", "174 | \n", "0 | \n", "0.0 | \n", "1 | \n", "1 | \n", "2 | \n", "0 | \n", "
303 rows × 14 columns
\n", "