C2_W4_Assignment__M.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "EZqu9a_ChWYv"
   },
   "source": [
    "# Cox Proportional Hazards and Random Survival Forests\n",
    "\n",
    "Welcome to the final assignment in Course 2! In this assignment you'll develop risk models using survival data and a combination of linear and non-linear techniques. We'll be using a dataset with survival data of patients with Primary Biliary Cirrhosis (pbc). PBC is a progressive disease of the liver caused by a buildup of bile within the liver (cholestasis) that results in damage to the small bile ducts that drain bile from the liver. Our goal will be to understand the effects of different factors on the survival times of the patients. Along the way you'll learn about the following topics: \n",
    "\n",
    "- Cox Proportional Hazards\n",
    "  - Data Preprocessing for Cox Models.\n",
    "- Random Survival Forests\n",
    "  - Permutation Methods for Interpretation."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Table of Contents\n",
    "\n",
    "- [1. Import Packages](#1)\n",
    "- [2. Load the Dataset](#2)\n",
    "- [3. Explore the Dataset](#3)\n",
    "- [4. Cox Proportional Hazards](#4)\n",
    "    - [Exercise 1 - to_one_hot](#ex-1)\n",
    "- [5. Fitting and Interpreting a Cox Model](#5)\n",
    "- [6. Hazard ratio](#3)\n",
    "    - [Exercise 2 - hazard_ratio](#ex-2)\n",
    "- [7. Harrell's C-Index](#7)\n",
    "    - [Exercise 3 - harrell_c](#ex-3)\n",
    "- [8. Random Survival Forests](#8)\n",
    "- [9. Permutation Method for Interpretation](#9)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "IH0ukiNS3zG-"
   },
   "source": [
    "<a name='1'></a>\n",
    "## 1. Import Packages\n",
    "\n",
    "We'll first import all the packages that we need for this assignment. \n",
    "\n",
    "- `sklearn` is one of the most popular machine learning libraries.\n",
    "- `numpy` is the fundamental package for scientific computing in python.\n",
    "- `pandas` is what we'll use to manipulate our data.\n",
    "- `matplotlib` is a plotting library.\n",
    "- `lifelines` is an open-source survival analysis library."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "0JHzRJaQi_nU"
   },
   "outputs": [],
   "source": [
    "import sklearn\n",
    "import numpy as np\n",
    "import pandas as pd\n",
    "import matplotlib.pyplot as plt\n",
    "\n",
    "from lifelines import CoxPHFitter\n",
    "from lifelines.utils import concordance_index as cindex\n",
    "from sklearn.model_selection import train_test_split\n",
    "\n",
    "from util import load_data\n",
    "\n",
    "from public_tests import *"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "vZMwq0VfW5TW"
   },
   "source": [
    "<a name='2'></a>\n",
    "## 2. Load the Dataset\n",
    "\n",
    "Run the next cell to load the data."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "df = load_data('data/pbc.csv')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a name='3'></a>\n",
    "## 3. Explore the Dataset\n",
    "\n",
    "In the lecture videos `time` was in months, however in this assignment, `time` will be converted into years. Also notice that we have assigned a numeric value to `sex`, where `female = 0` and `male = 1`.\n",
    "\n",
    "Next, familiarize yourself with the data and the shape of it. "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 241
    },
    "colab_type": "code",
    "id": "T1a_aHGmXT_C",
    "outputId": "1bbcf6d9-f293-49f4-963a-827c8e79813b"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(258, 19)\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>time</th>\n",
       "      <th>status</th>\n",
       "      <th>trt</th>\n",
       "      <th>age</th>\n",
       "      <th>sex</th>\n",
       "      <th>ascites</th>\n",
       "      <th>hepato</th>\n",
       "      <th>spiders</th>\n",
       "      <th>edema</th>\n",
       "      <th>bili</th>\n",
       "      <th>chol</th>\n",
       "      <th>albumin</th>\n",
       "      <th>copper</th>\n",
       "      <th>alk.phos</th>\n",
       "      <th>ast</th>\n",
       "      <th>trig</th>\n",
       "      <th>platelet</th>\n",
       "      <th>protime</th>\n",
       "      <th>stage</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1.095890</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>58.765229</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>14.5</td>\n",
       "      <td>261.0</td>\n",
       "      <td>2.60</td>\n",
       "      <td>156.0</td>\n",
       "      <td>1718.0</td>\n",
       "      <td>137.95</td>\n",
       "      <td>172.0</td>\n",
       "      <td>190.0</td>\n",
       "      <td>12.2</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>12.328767</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>56.446270</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.1</td>\n",
       "      <td>302.0</td>\n",
       "      <td>4.14</td>\n",
       "      <td>54.0</td>\n",
       "      <td>7394.8</td>\n",
       "      <td>113.52</td>\n",
       "      <td>88.0</td>\n",
       "      <td>221.0</td>\n",
       "      <td>10.6</td>\n",
       "      <td>3.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2.772603</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>70.072553</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.5</td>\n",
       "      <td>1.4</td>\n",
       "      <td>176.0</td>\n",
       "      <td>3.48</td>\n",
       "      <td>210.0</td>\n",
       "      <td>516.0</td>\n",
       "      <td>96.10</td>\n",
       "      <td>55.0</td>\n",
       "      <td>151.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>5.273973</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>54.740589</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.5</td>\n",
       "      <td>1.8</td>\n",
       "      <td>244.0</td>\n",
       "      <td>2.54</td>\n",
       "      <td>64.0</td>\n",
       "      <td>6121.8</td>\n",
       "      <td>60.63</td>\n",
       "      <td>92.0</td>\n",
       "      <td>183.0</td>\n",
       "      <td>10.3</td>\n",
       "      <td>4.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>5.019178</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>55.534565</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>322.0</td>\n",
       "      <td>4.09</td>\n",
       "      <td>52.0</td>\n",
       "      <td>824.0</td>\n",
       "      <td>60.45</td>\n",
       "      <td>213.0</td>\n",
       "      <td>204.0</td>\n",
       "      <td>9.7</td>\n",
       "      <td>3.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "        time  status  trt        age  sex  ascites  hepato  spiders  edema  \\\n",
       "0   1.095890     1.0  0.0  58.765229  0.0      1.0     1.0      1.0    1.0   \n",
       "1  12.328767     0.0  0.0  56.446270  0.0      0.0     1.0      1.0    0.0   \n",
       "2   2.772603     1.0  0.0  70.072553  1.0      0.0     0.0      0.0    0.5   \n",
       "3   5.273973     1.0  0.0  54.740589  0.0      0.0     1.0      1.0    0.5   \n",
       "6   5.019178     0.0  1.0  55.534565  0.0      0.0     1.0      0.0    0.0   \n",
       "\n",
       "   bili   chol  albumin  copper  alk.phos     ast   trig  platelet  protime  \\\n",
       "0  14.5  261.0     2.60   156.0    1718.0  137.95  172.0     190.0     12.2   \n",
       "1   1.1  302.0     4.14    54.0    7394.8  113.52   88.0     221.0     10.6   \n",
       "2   1.4  176.0     3.48   210.0     516.0   96.10   55.0     151.0     12.0   \n",
       "3   1.8  244.0     2.54    64.0    6121.8   60.63   92.0     183.0     10.3   \n",
       "6   1.0  322.0     4.09    52.0     824.0   60.45  213.0     204.0      9.7   \n",
       "\n",
       "   stage  \n",
       "0    4.0  \n",
       "1    3.0  \n",
       "2    4.0  \n",
       "3    4.0  \n",
       "6    3.0  "
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "print(df.shape)\n",
    "\n",
    "# df.head() only outputs the top few rows\n",
    "df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "Zy5BmjCV-Uo2"
   },
   "source": [
    "Take a minute to examine particular cases."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 357
    },
    "colab_type": "code",
    "id": "01I3ChzL-T-f",
    "outputId": "68e209dc-7a44-434b-d44c-4a1e817ee6ca"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "time          11.175342\n",
       "status         1.000000\n",
       "trt            0.000000\n",
       "age           44.520192\n",
       "sex            1.000000\n",
       "ascites        0.000000\n",
       "hepato         1.000000\n",
       "spiders        0.000000\n",
       "edema          0.000000\n",
       "bili           2.100000\n",
       "chol         456.000000\n",
       "albumin        4.000000\n",
       "copper       124.000000\n",
       "alk.phos    5719.000000\n",
       "ast          221.880000\n",
       "trig         230.000000\n",
       "platelet      70.000000\n",
       "protime        9.900000\n",
       "stage          2.000000\n",
       "Name: 23, dtype: float64"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "i = 20\n",
    "df.iloc[i, :]"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "pYZKl_9Tk2vS"
   },
   "source": [
    "Now, split your dataset into train, validation and test set using 60/20/20 split. "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "V4HJSZaMk1xG"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Total number of patients: 258\n",
      "Total number of patients in training set: 154\n",
      "Total number of patients in validation set: 52\n",
      "Total number of patients in test set: 52\n"
     ]
    }
   ],
   "source": [
    "np.random.seed(0)\n",
    "df_dev, df_test = train_test_split(df, test_size = 0.2)\n",
    "df_train, df_val = train_test_split(df_dev, test_size = 0.25)\n",
    "\n",
    "print(\"Total number of patients:\", df.shape[0])\n",
    "print(\"Total number of patients in training set:\", df_train.shape[0])\n",
    "print(\"Total number of patients in validation set:\", df_val.shape[0])\n",
    "print(\"Total number of patients in test set:\", df_test.shape[0])"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Before proceeding to modeling, let's normalize the continuous covariates to make sure they're on the same scale. Again, we should normalize the test data using statistics from the train data."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "continuous_columns = ['age', 'bili', 'chol', 'albumin', 'copper', 'alk.phos', 'ast', 'trig', 'platelet', 'protime']\n",
    "mean = df_train.loc[:, continuous_columns].mean()\n",
    "std = df_train.loc[:, continuous_columns].std()\n",
    "df_train.loc[:, continuous_columns] = (df_train.loc[:, continuous_columns] - mean) / std\n",
    "df_val.loc[:, continuous_columns] = (df_val.loc[:, continuous_columns] - mean) / std\n",
    "df_test.loc[:, continuous_columns] = (df_test.loc[:, continuous_columns] - mean) / std"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Let's check the summary statistics on our training dataset to make sure it's standardized."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "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>bili</th>\n",
       "      <th>chol</th>\n",
       "      <th>albumin</th>\n",
       "      <th>copper</th>\n",
       "      <th>alk.phos</th>\n",
       "      <th>ast</th>\n",
       "      <th>trig</th>\n",
       "      <th>platelet</th>\n",
       "      <th>protime</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>count</th>\n",
       "      <td>1.540000e+02</td>\n",
       "      <td>1.540000e+02</td>\n",
       "      <td>1.540000e+02</td>\n",
       "      <td>1.540000e+02</td>\n",
       "      <td>1.540000e+02</td>\n",
       "      <td>1.540000e+02</td>\n",
       "      <td>1.540000e+02</td>\n",
       "      <td>1.540000e+02</td>\n",
       "      <td>1.540000e+02</td>\n",
       "      <td>1.540000e+02</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>mean</th>\n",
       "      <td>9.833404e-16</td>\n",
       "      <td>-3.258577e-16</td>\n",
       "      <td>1.153478e-16</td>\n",
       "      <td>1.153478e-16</td>\n",
       "      <td>5.767392e-18</td>\n",
       "      <td>1.326500e-16</td>\n",
       "      <td>-1.263059e-15</td>\n",
       "      <td>8.074349e-17</td>\n",
       "      <td>2.018587e-17</td>\n",
       "      <td>1.291896e-14</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>std</th>\n",
       "      <td>1.000000e+00</td>\n",
       "      <td>1.000000e+00</td>\n",
       "      <td>1.000000e+00</td>\n",
       "      <td>1.000000e+00</td>\n",
       "      <td>1.000000e+00</td>\n",
       "      <td>1.000000e+00</td>\n",
       "      <td>1.000000e+00</td>\n",
       "      <td>1.000000e+00</td>\n",
       "      <td>1.000000e+00</td>\n",
       "      <td>1.000000e+00</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>min</th>\n",
       "      <td>-2.304107e+00</td>\n",
       "      <td>-5.735172e-01</td>\n",
       "      <td>-1.115330e+00</td>\n",
       "      <td>-3.738104e+00</td>\n",
       "      <td>-9.856552e-01</td>\n",
       "      <td>-7.882167e-01</td>\n",
       "      <td>-1.489281e+00</td>\n",
       "      <td>-1.226674e+00</td>\n",
       "      <td>-2.058899e+00</td>\n",
       "      <td>-1.735556e+00</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25%</th>\n",
       "      <td>-6.535035e-01</td>\n",
       "      <td>-4.895812e-01</td>\n",
       "      <td>-5.186963e-01</td>\n",
       "      <td>-5.697976e-01</td>\n",
       "      <td>-6.470611e-01</td>\n",
       "      <td>-5.186471e-01</td>\n",
       "      <td>-8.353982e-01</td>\n",
       "      <td>-6.884514e-01</td>\n",
       "      <td>-6.399831e-01</td>\n",
       "      <td>-7.382590e-01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>50%</th>\n",
       "      <td>-6.443852e-03</td>\n",
       "      <td>-3.846612e-01</td>\n",
       "      <td>-2.576693e-01</td>\n",
       "      <td>5.663556e-02</td>\n",
       "      <td>-3.140636e-01</td>\n",
       "      <td>-3.416086e-01</td>\n",
       "      <td>-2.260984e-01</td>\n",
       "      <td>-2.495932e-01</td>\n",
       "      <td>-4.100373e-02</td>\n",
       "      <td>-1.398807e-01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>75%</th>\n",
       "      <td>5.724289e-01</td>\n",
       "      <td>2.977275e-02</td>\n",
       "      <td>1.798617e-01</td>\n",
       "      <td>6.890921e-01</td>\n",
       "      <td>3.435366e-01</td>\n",
       "      <td>-4.620597e-03</td>\n",
       "      <td>6.061159e-01</td>\n",
       "      <td>3.755727e-01</td>\n",
       "      <td>6.617988e-01</td>\n",
       "      <td>3.587680e-01</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>max</th>\n",
       "      <td>2.654276e+00</td>\n",
       "      <td>5.239050e+00</td>\n",
       "      <td>6.243146e+00</td>\n",
       "      <td>2.140730e+00</td>\n",
       "      <td>5.495204e+00</td>\n",
       "      <td>4.869263e+00</td>\n",
       "      <td>3.058176e+00</td>\n",
       "      <td>5.165751e+00</td>\n",
       "      <td>3.190823e+00</td>\n",
       "      <td>4.447687e+00</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                age          bili          chol       albumin        copper  \\\n",
       "count  1.540000e+02  1.540000e+02  1.540000e+02  1.540000e+02  1.540000e+02   \n",
       "mean   9.833404e-16 -3.258577e-16  1.153478e-16  1.153478e-16  5.767392e-18   \n",
       "std    1.000000e+00  1.000000e+00  1.000000e+00  1.000000e+00  1.000000e+00   \n",
       "min   -2.304107e+00 -5.735172e-01 -1.115330e+00 -3.738104e+00 -9.856552e-01   \n",
       "25%   -6.535035e-01 -4.895812e-01 -5.186963e-01 -5.697976e-01 -6.470611e-01   \n",
       "50%   -6.443852e-03 -3.846612e-01 -2.576693e-01  5.663556e-02 -3.140636e-01   \n",
       "75%    5.724289e-01  2.977275e-02  1.798617e-01  6.890921e-01  3.435366e-01   \n",
       "max    2.654276e+00  5.239050e+00  6.243146e+00  2.140730e+00  5.495204e+00   \n",
       "\n",
       "           alk.phos           ast          trig      platelet       protime  \n",
       "count  1.540000e+02  1.540000e+02  1.540000e+02  1.540000e+02  1.540000e+02  \n",
       "mean   1.326500e-16 -1.263059e-15  8.074349e-17  2.018587e-17  1.291896e-14  \n",
       "std    1.000000e+00  1.000000e+00  1.000000e+00  1.000000e+00  1.000000e+00  \n",
       "min   -7.882167e-01 -1.489281e+00 -1.226674e+00 -2.058899e+00 -1.735556e+00  \n",
       "25%   -5.186471e-01 -8.353982e-01 -6.884514e-01 -6.399831e-01 -7.382590e-01  \n",
       "50%   -3.416086e-01 -2.260984e-01 -2.495932e-01 -4.100373e-02 -1.398807e-01  \n",
       "75%   -4.620597e-03  6.061159e-01  3.755727e-01  6.617988e-01  3.587680e-01  \n",
       "max    4.869263e+00  3.058176e+00  5.165751e+00  3.190823e+00  4.447687e+00  "
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df_train.loc[:, continuous_columns].describe()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "BX3woHz-jit1"
   },
   "source": [
    "<a name='4'></a>\n",
    "## 4. Cox Proportional Hazards\n",
    "\n",
    "Our goal is to build a risk score using the survival data that we have. We'll begin by fitting a Cox Proportional Hazards model to your data.\n",
    "\n",
    "Recall that the Cox Proportional Hazards model describes the hazard for an individual $i$ at time $t$ as \n",
    "\n",
    "$$\n",
    "\\lambda(t, x) = \\lambda_0(t)e^{\\theta^T X_i}\n",
    "$$\n",
    "\n",
    "The $\\lambda_0$ term is a baseline hazard and incorporates the risk over time, and the other term incorporates the risk due to the individual's covariates. After fitting the model, we can rank individuals using the person-dependent risk term $e^{\\theta^T X_i}$. \n",
    "\n",
    "Categorical variables cannot be used in a regression model as they are. In order to use them, conversion to a series of variables is required.\n",
    "\n",
    "Since our data has a mix of categorical (`stage`) and continuous (`wblc`) variables, before we proceed further we need to do some data engineering. To tackle the issue at hand we'll be using the `Dummy Coding` technique. In order to use Cox Proportional Hazards, we will have to turn the categorical data into one hot features so that we can fit our Cox model. Luckily, Pandas has a built-in function called `get_dummies` that will make it easier for us to implement our function. It turns categorical features into multiple binary features.\n",
    "\n",
    "<img src=\"images/1-hot-encode.png\" style=\"padding-top: 5px;width: 60%;left: 0px;margin-left: 150px;margin-right: 0px;\">\n",
    "\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a name='ex-1'></a>\n",
    "### Exercise 1 - to_one_hot\n",
    "In the cell below, implement the `to_one_hot(...)` function."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<details>    \n",
    "<summary>\n",
    "    <font size=\"3\" color=\"darkgreen\"><b>Hints</b></font>\n",
    "</summary>\n",
    "<p>\n",
    "<ul>\n",
    "    <li>Remember to drop the first dummy for each each category to avoid convergence issues when fitting the proportional hazards model.</li>\n",
    "    <li> Check out the <a href=\"https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.get_dummies.html\" > get_dummies() </a>  documentation. </li>\n",
    "    <li>Use <code>dtype=np.float64</code>.</li>\n",
    "</ul>\n",
    "</p>"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "VMzvx0xF_C3I"
   },
   "outputs": [],
   "source": [
    "# UNQ_C1 (UNIQUE CELL IDENTIFIER, DO NOT EDIT)\n",
    "def to_one_hot(dataframe, columns):\n",
    "    '''\n",
    "    Convert columns in dataframe to one-hot encoding.\n",
    "    Args:\n",
    "        dataframe (dataframe): pandas dataframe containing covariates\n",
    "        columns (list of strings): list categorical column names to one hot encode\n",
    "    Returns:\n",
    "        one_hot_df (dataframe): dataframe with categorical columns encoded\n",
    "                            as binary variables\n",
    "    '''\n",
    "    \n",
    "    ### START CODE HERE (REPLACE INSTANCES OF 'None' with your code) ###\n",
    "    \n",
    "    one_hot_df = pd.get_dummies(data=dataframe, columns=columns, drop_first=True, dtype=np.float64)\n",
    "    \n",
    "    ### END CODE HERE ###\n",
    "    \n",
    "    return one_hot_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "SGZfLeup_fUL"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "One hot val columns:\n",
      "\n",
      " ['time', 'status', 'trt', 'age', 'sex', 'ascites', 'hepato', 'edema', 'bili', 'chol', 'albumin', 'copper', 'alk.phos', 'ast', 'trig', 'platelet', 'protime', 'spiders_1.0', 'stage_2.0', 'stage_3.0', 'stage_4.0'] \n",
      "\n",
      "There are 21 columns\n",
      "\n",
      "\u001b[92m All tests passed.\n"
     ]
    }
   ],
   "source": [
    "# test cell ex1 - do not modify this test cell\n",
    "to_one_hot_test(to_one_hot, df_train, df_val, df_test)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Expected Output:\n",
    "```\n",
    "One hot val columns:\n",
    "\n",
    " ['time', 'status', 'trt', 'age', 'sex', 'ascites', 'hepato', 'edema', 'bili', 'chol', 'albumin', 'copper', 'alk.phos', 'ast', 'trig', 'platelet', 'protime', 'spiders_1.0', 'stage_2.0', 'stage_3.0', 'stage_4.0'] \n",
    "\n",
    "There are 21 columns\n",
    "\n",
    " All tests passed.\n",
    "```"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Now you'll use the function you coded to transform the training, validation, and test sets."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "to_encode = ['edema', 'stage']\n",
    "\n",
    "one_hot_train = to_one_hot(df_train, to_encode)\n",
    "one_hot_val = to_one_hot(df_val, to_encode)\n",
    "one_hot_test = to_one_hot(df_test, to_encode)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Look for new features\n",
    "Now, let's take a peek at one of the transformed data sets. Do you notice any new features?"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 241
    },
    "colab_type": "code",
    "id": "w8EG8A9gXcpu",
    "outputId": "384d9ade-2c96-4979-d3b7-da2b8e50f2e0"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(154, 22)\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>time</th>\n",
       "      <th>status</th>\n",
       "      <th>trt</th>\n",
       "      <th>age</th>\n",
       "      <th>sex</th>\n",
       "      <th>ascites</th>\n",
       "      <th>hepato</th>\n",
       "      <th>spiders</th>\n",
       "      <th>bili</th>\n",
       "      <th>chol</th>\n",
       "      <th>...</th>\n",
       "      <th>alk.phos</th>\n",
       "      <th>ast</th>\n",
       "      <th>trig</th>\n",
       "      <th>platelet</th>\n",
       "      <th>protime</th>\n",
       "      <th>edema_0.5</th>\n",
       "      <th>edema_1.0</th>\n",
       "      <th>stage_2.0</th>\n",
       "      <th>stage_3.0</th>\n",
       "      <th>stage_4.0</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>279</th>\n",
       "      <td>3.868493</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>-0.414654</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>-0.300725</td>\n",
       "      <td>-0.096081</td>\n",
       "      <td>...</td>\n",
       "      <td>0.167937</td>\n",
       "      <td>0.401418</td>\n",
       "      <td>0.330031</td>\n",
       "      <td>0.219885</td>\n",
       "      <td>-1.137178</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>137</th>\n",
       "      <td>3.553425</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.069681</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.895363</td>\n",
       "      <td>0.406085</td>\n",
       "      <td>...</td>\n",
       "      <td>0.101665</td>\n",
       "      <td>0.472367</td>\n",
       "      <td>1.621764</td>\n",
       "      <td>-0.120868</td>\n",
       "      <td>-0.239610</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>249</th>\n",
       "      <td>4.846575</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>-0.924494</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>-0.510565</td>\n",
       "      <td>-0.225352</td>\n",
       "      <td>...</td>\n",
       "      <td>0.245463</td>\n",
       "      <td>1.899020</td>\n",
       "      <td>-0.580807</td>\n",
       "      <td>0.422207</td>\n",
       "      <td>0.159309</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>266</th>\n",
       "      <td>0.490411</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.938314</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.748475</td>\n",
       "      <td>-0.608191</td>\n",
       "      <td>...</td>\n",
       "      <td>-0.650254</td>\n",
       "      <td>-0.288898</td>\n",
       "      <td>-0.481443</td>\n",
       "      <td>-0.727833</td>\n",
       "      <td>1.356065</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>12.328767</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.563645</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>-0.405645</td>\n",
       "      <td>-0.210436</td>\n",
       "      <td>...</td>\n",
       "      <td>2.173526</td>\n",
       "      <td>-0.144699</td>\n",
       "      <td>-0.531125</td>\n",
       "      <td>-0.450972</td>\n",
       "      <td>-0.139881</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>5 rows × 22 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "          time  status  trt       age  sex  ascites  hepato  spiders  \\\n",
       "279   3.868493     0.0  0.0 -0.414654  0.0      0.0     0.0      0.0   \n",
       "137   3.553425     1.0  0.0  0.069681  1.0      0.0     1.0      0.0   \n",
       "249   4.846575     0.0  1.0 -0.924494  0.0      0.0     1.0      0.0   \n",
       "266   0.490411     1.0  0.0  1.938314  0.0      1.0     1.0      1.0   \n",
       "1    12.328767     0.0  0.0  0.563645  0.0      0.0     1.0      1.0   \n",
       "\n",
       "         bili      chol  ...  alk.phos       ast      trig  platelet  \\\n",
       "279 -0.300725 -0.096081  ...  0.167937  0.401418  0.330031  0.219885   \n",
       "137  0.895363  0.406085  ...  0.101665  0.472367  1.621764 -0.120868   \n",
       "249 -0.510565 -0.225352  ...  0.245463  1.899020 -0.580807  0.422207   \n",
       "266  0.748475 -0.608191  ... -0.650254 -0.288898 -0.481443 -0.727833   \n",
       "1   -0.405645 -0.210436  ...  2.173526 -0.144699 -0.531125 -0.450972   \n",
       "\n",
       "      protime  edema_0.5  edema_1.0  stage_2.0  stage_3.0  stage_4.0  \n",
       "279 -1.137178        0.0        0.0        0.0        1.0        0.0  \n",
       "137 -0.239610        0.0        0.0        0.0        1.0        0.0  \n",
       "249  0.159309        0.0        0.0        0.0        0.0        1.0  \n",
       "266  1.356065        0.0        1.0        0.0        0.0        1.0  \n",
       "1   -0.139881        0.0        0.0        0.0        1.0        0.0  \n",
       "\n",
       "[5 rows x 22 columns]"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "print(one_hot_train.shape)\n",
    "one_hot_train.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "hNxuymLwyjqM"
   },
   "source": [
    "<a name='5'></a>\n",
    "## 5. Fitting and Interpreting a Cox Model"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "ygiFcUKcAFQk"
   },
   "source": [
    "Run the following cell to fit your Cox Proportional Hazards model using the `lifelines` package."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 34
    },
    "colab_type": "code",
    "id": "dDCS7p3xjbXB",
    "outputId": "41b12f82-8b35-43e1-d2a9-05258ac50b20"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<lifelines.CoxPHFitter: fitted with 154 total observations, 90 right-censored observations>"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "cph = CoxPHFitter()\n",
    "cph.fit(one_hot_train, duration_col = 'time', event_col = 'status', step_size=0.1)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "5MUITR0QANDH"
   },
   "source": [
    "You can use `cph.print_summary()` to view the coefficients associated with each covariate as well as confidence intervals. "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 903
    },
    "colab_type": "code",
    "id": "fH5AZs8vjcEv",
    "outputId": "5429f7d5-5669-431f-a014-cf609c90997f"
   },
   "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",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>model</th>\n",
       "      <td>lifelines.CoxPHFitter</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>duration col</th>\n",
       "      <td>'time'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>event col</th>\n",
       "      <td>'status'</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>number of observations</th>\n",
       "      <td>154</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>number of events observed</th>\n",
       "      <td>64</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>partial log-likelihood</th>\n",
       "      <td>-230.82</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>time fit was run</th>\n",
       "      <td>2024-10-27 09:20:11 UTC</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div><table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>coef</th>\n",
       "      <th>exp(coef)</th>\n",
       "      <th>se(coef)</th>\n",
       "      <th>coef lower 95%</th>\n",
       "      <th>coef upper 95%</th>\n",
       "      <th>exp(coef) lower 95%</th>\n",
       "      <th>exp(coef) upper 95%</th>\n",
       "      <th>z</th>\n",
       "      <th>p</th>\n",
       "      <th>-log2(p)</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>trt</th>\n",
       "      <td>-0.22</td>\n",
       "      <td>0.80</td>\n",
       "      <td>0.30</td>\n",
       "      <td>-0.82</td>\n",
       "      <td>0.37</td>\n",
       "      <td>0.44</td>\n",
       "      <td>1.45</td>\n",
       "      <td>-0.73</td>\n",
       "      <td>0.46</td>\n",
       "      <td>1.11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>age</th>\n",
       "      <td>0.23</td>\n",
       "      <td>1.26</td>\n",
       "      <td>0.19</td>\n",
       "      <td>-0.13</td>\n",
       "      <td>0.60</td>\n",
       "      <td>0.88</td>\n",
       "      <td>1.82</td>\n",
       "      <td>1.26</td>\n",
       "      <td>0.21</td>\n",
       "      <td>2.27</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>sex</th>\n",
       "      <td>0.34</td>\n",
       "      <td>1.41</td>\n",
       "      <td>0.40</td>\n",
       "      <td>-0.45</td>\n",
       "      <td>1.14</td>\n",
       "      <td>0.64</td>\n",
       "      <td>3.11</td>\n",
       "      <td>0.84</td>\n",
       "      <td>0.40</td>\n",
       "      <td>1.33</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ascites</th>\n",
       "      <td>-0.10</td>\n",
       "      <td>0.91</td>\n",
       "      <td>0.56</td>\n",
       "      <td>-1.20</td>\n",
       "      <td>1.01</td>\n",
       "      <td>0.30</td>\n",
       "      <td>2.75</td>\n",
       "      <td>-0.17</td>\n",
       "      <td>0.86</td>\n",
       "      <td>0.21</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>hepato</th>\n",
       "      <td>0.31</td>\n",
       "      <td>1.36</td>\n",
       "      <td>0.38</td>\n",
       "      <td>-0.44</td>\n",
       "      <td>1.06</td>\n",
       "      <td>0.64</td>\n",
       "      <td>2.89</td>\n",
       "      <td>0.81</td>\n",
       "      <td>0.42</td>\n",
       "      <td>1.26</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>spiders</th>\n",
       "      <td>-0.18</td>\n",
       "      <td>0.83</td>\n",
       "      <td>0.38</td>\n",
       "      <td>-0.94</td>\n",
       "      <td>0.57</td>\n",
       "      <td>0.39</td>\n",
       "      <td>1.77</td>\n",
       "      <td>-0.47</td>\n",
       "      <td>0.64</td>\n",
       "      <td>0.66</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>bili</th>\n",
       "      <td>0.05</td>\n",
       "      <td>1.05</td>\n",
       "      <td>0.18</td>\n",
       "      <td>-0.29</td>\n",
       "      <td>0.39</td>\n",
       "      <td>0.75</td>\n",
       "      <td>1.48</td>\n",
       "      <td>0.29</td>\n",
       "      <td>0.77</td>\n",
       "      <td>0.37</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>chol</th>\n",
       "      <td>0.19</td>\n",
       "      <td>1.20</td>\n",
       "      <td>0.15</td>\n",
       "      <td>-0.10</td>\n",
       "      <td>0.47</td>\n",
       "      <td>0.91</td>\n",
       "      <td>1.60</td>\n",
       "      <td>1.28</td>\n",
       "      <td>0.20</td>\n",
       "      <td>2.33</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>albumin</th>\n",
       "      <td>-0.40</td>\n",
       "      <td>0.67</td>\n",
       "      <td>0.18</td>\n",
       "      <td>-0.75</td>\n",
       "      <td>-0.06</td>\n",
       "      <td>0.47</td>\n",
       "      <td>0.94</td>\n",
       "      <td>-2.28</td>\n",
       "      <td>0.02</td>\n",
       "      <td>5.46</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>copper</th>\n",
       "      <td>0.30</td>\n",
       "      <td>1.35</td>\n",
       "      <td>0.16</td>\n",
       "      <td>-0.01</td>\n",
       "      <td>0.61</td>\n",
       "      <td>0.99</td>\n",
       "      <td>1.84</td>\n",
       "      <td>1.91</td>\n",
       "      <td>0.06</td>\n",
       "      <td>4.14</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>alk.phos</th>\n",
       "      <td>-0.22</td>\n",
       "      <td>0.80</td>\n",
       "      <td>0.14</td>\n",
       "      <td>-0.49</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.61</td>\n",
       "      <td>1.05</td>\n",
       "      <td>-1.62</td>\n",
       "      <td>0.11</td>\n",
       "      <td>3.24</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ast</th>\n",
       "      <td>0.21</td>\n",
       "      <td>1.24</td>\n",
       "      <td>0.16</td>\n",
       "      <td>-0.10</td>\n",
       "      <td>0.53</td>\n",
       "      <td>0.91</td>\n",
       "      <td>1.69</td>\n",
       "      <td>1.34</td>\n",
       "      <td>0.18</td>\n",
       "      <td>2.48</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>trig</th>\n",
       "      <td>0.20</td>\n",
       "      <td>1.23</td>\n",
       "      <td>0.16</td>\n",
       "      <td>-0.11</td>\n",
       "      <td>0.52</td>\n",
       "      <td>0.89</td>\n",
       "      <td>1.68</td>\n",
       "      <td>1.27</td>\n",
       "      <td>0.21</td>\n",
       "      <td>2.28</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>platelet</th>\n",
       "      <td>0.14</td>\n",
       "      <td>1.15</td>\n",
       "      <td>0.15</td>\n",
       "      <td>-0.16</td>\n",
       "      <td>0.43</td>\n",
       "      <td>0.86</td>\n",
       "      <td>1.54</td>\n",
       "      <td>0.92</td>\n",
       "      <td>0.36</td>\n",
       "      <td>1.48</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>protime</th>\n",
       "      <td>0.36</td>\n",
       "      <td>1.43</td>\n",
       "      <td>0.17</td>\n",
       "      <td>0.03</td>\n",
       "      <td>0.69</td>\n",
       "      <td>1.03</td>\n",
       "      <td>1.99</td>\n",
       "      <td>2.15</td>\n",
       "      <td>0.03</td>\n",
       "      <td>4.97</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>edema_0.5</th>\n",
       "      <td>1.24</td>\n",
       "      <td>3.47</td>\n",
       "      <td>0.46</td>\n",
       "      <td>0.35</td>\n",
       "      <td>2.14</td>\n",
       "      <td>1.42</td>\n",
       "      <td>8.50</td>\n",
       "      <td>2.72</td>\n",
       "      <td>0.01</td>\n",
       "      <td>7.28</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>edema_1.0</th>\n",
       "      <td>2.02</td>\n",
       "      <td>7.51</td>\n",
       "      <td>0.60</td>\n",
       "      <td>0.84</td>\n",
       "      <td>3.20</td>\n",
       "      <td>2.31</td>\n",
       "      <td>24.43</td>\n",
       "      <td>3.35</td>\n",
       "      <td>&lt;0.005</td>\n",
       "      <td>10.28</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>stage_2.0</th>\n",
       "      <td>1.21</td>\n",
       "      <td>3.35</td>\n",
       "      <td>1.08</td>\n",
       "      <td>-0.92</td>\n",
       "      <td>3.33</td>\n",
       "      <td>0.40</td>\n",
       "      <td>28.06</td>\n",
       "      <td>1.11</td>\n",
       "      <td>0.27</td>\n",
       "      <td>1.91</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>stage_3.0</th>\n",
       "      <td>1.18</td>\n",
       "      <td>3.27</td>\n",
       "      <td>1.09</td>\n",
       "      <td>-0.96</td>\n",
       "      <td>3.33</td>\n",
       "      <td>0.38</td>\n",
       "      <td>27.86</td>\n",
       "      <td>1.08</td>\n",
       "      <td>0.28</td>\n",
       "      <td>1.84</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>stage_4.0</th>\n",
       "      <td>1.41</td>\n",
       "      <td>4.10</td>\n",
       "      <td>1.15</td>\n",
       "      <td>-0.85</td>\n",
       "      <td>3.67</td>\n",
       "      <td>0.43</td>\n",
       "      <td>39.43</td>\n",
       "      <td>1.22</td>\n",
       "      <td>0.22</td>\n",
       "      <td>2.18</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table><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",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>Concordance</th>\n",
       "      <td>0.83</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Log-likelihood ratio test</th>\n",
       "      <td>97.63 on 20 df, -log2(p)=38.13</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "<IPython.core.display.HTML object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "cph.print_summary()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "**Question:**\n",
    "\n",
    "- According to the model, was treatment `trt` beneficial? \n",
    "- What was its associated hazard ratio? \n",
    "    - Note that the hazard ratio is how much an incremental increase in the feature variable changes the hazard."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<details>    \n",
    "<summary>\n",
    "    <font size=\"3\" color=\"darkgreen\"><b>Check your answer!</b></font>\n",
    "</summary>\n",
    "<p>\n",
    "<ul>\n",
    "<ul>\n",
    "    <li>You should see that the treatment (trt) was beneficial because it has a negative impact on the hazard (the coefficient is negative, and exp(coef) is less than 1).</li>\n",
    "    <li>The associated hazard ratio is ~0.8, because this is the exp(coef) of treatment.</li>\n",
    "</ul>\n",
    "</p>"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "We can compare the predicted survival curves for treatment variables. Run the next cell to plot survival curves using the `plot_covariate_groups()` function. \n",
    "- The y-axis is th survival rate\n",
    "- The x-axis is time"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 282
    },
    "colab_type": "code",
    "id": "Uxl0icyBS4Dr",
    "outputId": "5fa08369-e89e-424f-f9f0-60cf7a1cfbcd"
   },
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAXQAAAD4CAYAAAD8Zh1EAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4xLjIsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy8li6FKAAAgAElEQVR4nO3df1yUZb7/8deV2qrlr6TSFfyVRpp6sDDWbHMq3DQNKy2IbBPb6FuaWdlJ9+SPrE062jnqZj+0Fd08rrZqpVZqbqKZZlBLK6KWaSmupmhqairG9f1jYBxggAFmGGZ4Px8PHnHPfXPPZ9I+XVz35/pcxlqLiIgEvwsCHYCIiPiGErqISIhQQhcRCRFK6CIiIUIJXUQkRNQN1BuHhYXZtm3bBurtRUSC0hdffJFrrb3U07mAJfS2bduSkZERqLcXEQlKxpjvSzunKRcRkRChhC4iEiKU0EVEQkTA5tBFpGry8vLIycnh9OnTgQ5F/KB+/fqEh4dTr149r39GCV0kSOXk5NCoUSPatm2LMSbQ4YgPWWs5fPgwOTk5tGvXzuufK3fKxRgzxxhz0BiTVcp5Y4yZYYzZaYz5lzHmmgrELSKVdPr0aZo3b65kHoKMMTRv3rzCv315M4c+F+hbxvl+QMeCr2TgtQpFICKVpmQeuirzZ1tuQrfWrgeOlHHJQOCv1ukzoKkxpmV599225Z+Mffhu7yMVEZEy+aLKpRWw1+04p+C1EowxycaYDGNMxi+/5BP5wwoyn7+eHu2bkHLPlWz++8s+CEdEqsvRo0d59dVXK3yuNLt37yYmJoYOHToQHx/P2bNnfRFmrVGtZYvW2lnW2mhrbXS7NuH0uO5a17nwc3vpmvkcPdo30chdJEiUlrTPnTtXqYT+zDPP8MQTT7Bz506aNWvGX/7yF1+FWiv4osplHxDhdhxe8FqZLmp2OVf/cQMA6eNg899fJnvrYs79/DlRh9/n2MzfMfDPXzAyti13XduC3J/OsvTAr0l+cZ4PQhYRXxgzZgzffvstUVFR1KtXj/r169OsWTO2b9/ONddc4zrXp08fpkyZUua9rLV8/PHHLFiwAIAHHniAiRMn8sgjj1THRwkJvkjoy4ARxpiFQAxwzFq7v6I3ibn7Kbj7KR5pMo72x9aw7YfjnDx7jj0/nmLr/mOkrkhnU84vJHfMha6DITrJB6GLhIbnlm8l+9/HfXrPzr9uzITbry7zmpSUFLKyssjMzCQtLY3+/fuTlZVFu3bt+O6771znAH766Sd++9vferzPggULuOyyy2jatCl16zrTUnh4OPv2lTs2FDflJnRjzN8ABxBmjMkBJgD1AKy1rwMfALcBO4FTQJUy7cXXP8TUzNsAaP8MbML5dbzpOO5ovZcDW9fD1vW02LJYiV2khrnuuutKrZtu1KiRK7l7kpub66+wao1yE7q19t5yzltguK8CSoxpTWJM6xKvL4iaw3uZ+/jzi/cQZo7x5cUb2Puv9dw/ZDTPDuhA7NVhJW+mhC+1RHkj6epy0UUXlXquvBF6p06dOHr0KOfOnaNu3brk5OTQqpXH+gopRdCsFC1M9H/4bjhfn/wVY/P28pv8tZw8u4vvjpzk8911+M+3d/DcHR3pHdmcHV+sY93Sj0m+a/H5myjBi/hUo0aN+Omnn7w6V94IHeCmm25i8eLFJCQkMG/ePAYOHOjTeENd0CT0Qm9OHgPAgs17WJB5N+07wwu7j5B38AB5+TOZmPd7Ljvbnb2f59Do53+TDGTl/MSIv2wksatbgldyF6my5s2b06tXL7p06UKDBg24/PLLPZ7r169fuQ9FAV566SUSEhJ49tln6d69Ow8++KA/ww85xjljUv2io6Otrza4WLB5D+9lFn14suadBdQLa0Pv3/bi2L5vOfjun3ixz8W0aFyfUambSOxaj+S7bj7/A0rwEmS2bdtGp06dAh2G+JGnP2NjzBfW2mhP1wfdCN0TT/PuC6JauZJ8k1ZXsP22qfw/4Mq8XE42O8IVnS4iffdRnn57O6/cfIYu3ztLKJXURSRYhURC96R4kj8/ir+Er2/9bx4EOuYdIDt/Jn+/uAfH984n67lHNecuIkErZBN6ce4J3j253/TUTObtPsKCT9ZwbbNTJAPrdhxmwsKMonPuxSnZi0gNU2sSujtPo/fHMm/jk4guxJ/txsG8f/JLy7NEda/Pmq25vLBiJ2899B9EXNKAlVsOkbL0Sz1gFZEap1Ym9OISY1rDc0+75twvu7I72Y0m83LLxjx0xUnIeAHumQcREbByJVH7XyYuxrI88wf279pK8vcbYIumakQksJTQCxQftce/sYns/ceZTWMuv3cyoz/IwdlIsgk4JvFxVCtm/v1eerbrDW3qnL/R9xucX6CkLiLVSptEl2JgVCs6t2zs8dzm3Uf44ztbONPuBjren8LSJg/imHeSY3ctgAHTnBetGAWp/Yt+ZaRW4ycQ8T9ft8995ZVX6NChA8YYtQKoBI3QS1FaCwI4/1B1Mzcyee0+7mx6yHVu0bcNSflbUx75TWOS27j9kEbuEoIKk/ajjz5a5HX39rnFz5WlV69eDBgwAIfD4eNIa4eQWFgUKAs27+GP72wBIKbdJQB8+8m7tP7Vaeb/eTJLly7lzTff5KOPPqLeV/Odo3aANjecv4nm26WSasLCooSEBN577z0iIyM9ts8tPOdN+1x3bdu2JSMjg7AwDz2aapFaubAoUApH8O6rVHPDbyQXeHdL0V8X52YZUuY2ZPsEtz20C0ftWzyURirRS0V8OAYObPHtPVt0hX4pZV7iy/a5nTt39m38tZASehV5KoH84ztb+OM7W4hpdwWX3zuZIXMy+PaTbVxwSVtIep/Zs2ezaNEi1qRM85zMleglSFWlfa5UnRK6j3katYNz5E74jSzYvOf8i9FJvPr5zyxfvpwPP/wQgOnTp/OPj//NsqRfl7y55uGlNOWMpKtLVdrnaoRedUrofuCxt0yRkXsXmt/dhfg3NtFw95GSN2gaAUnLmDp1Kps2bWLJkiUApIwYzCXfv09ym8VK6FIj+Lp9rlSNyharSWJMa168s6vr4SlA9v7jnOpwi2t0DvD444+zbNkyj/eYuyaLjw82c86VqhRSagD3FrlPP/201+dKM2PGDMLDw8nJyaFbt2784Q9/8EfYIUtVLgFUuHjJU737wKhWnssmM1LPz60f2OJ8cJX0vp8jlZqoJlS5iH+pyiWIDIzyvL3W5t1H2Lz7SIl5eID8L7PoFv4bxo0b5xyhF47WPdEDVJFaRQk9gErdP9XDhh3gTPS5G75kf0RTxgHjP61DeF4Tklt4uHlZlTLFKfGLhAQl9BqorET/R0bTqmAe/u0NX9OzZ0+Sk1IZO3Yshw8fZtasWQCMHtKXK+vsK7pa1ZPC2mUldJGgp4QeRBJjWhcZuW/fvr3Ua1dkfMctt9wCSTPLvqmnaRuN2EWCkhJ6EMref5z4NzYB5x+eTp48ucg1hcl+1Chnu4Fp05xNw4YPH06DBg2YOnUqAMnLThDd2G3aRiN2kaClhB5k3B+kZu8/DlBqEzFvrN/2A83vvBOSCv6HoBG7SNBS2WIQK6vsEcooffQgKSmJiIgIJsW1K/ogVaWRNVagyxa/++47BgwYQFZWll/un5aWxtSpU1mxYgXLli0jOzubMWPG+OW9qiojI4O//vWvzJgxo8I/O3fuXDIyMnjllVdKnFPZYi1SWtkjlF36WOZ9opOKjsaLj9g1WpcAiIuLIy4uLqAxnDt3jrp1PafM6OhooqM95thqpZWiQSwxpjWLHu7p8av4qtTSZO8/znuZ+0hNTWXSpEkMGTKEq666ynU+4W+5zNrexHlwYIt3ZZBSa5w7d4777ruPTp06MXjwYE6dOgXApEmT6NGjB126dCE5OZnCmYAZM2bQuXNnunXrRkJCAgAnT55k2LBhXHfddXTv3p333nuvxPvMnTuXESNGADB06FBGjhzJ9ddfT/v27Vm8+PzfySlTptCjRw+6devGhAkTStznl19+YejQoXTp0oWuXbvyv//7vwA4HA4KZwxyc3Np27at633j4uK4+eabueWWW0hISOD998//tjp06FAWL15MWloaAwYMID8/n7Zt23L06FHXNR07duSHH35g+fLlxMTE0L17d2JjY/nhhx8q/e+9VNbagHxde+21VgLvntc32i4TVtp7Xt9o73l9o732vv+0g5OfdJ2PjIy0kydPdh7Muc3aFyOc/yz+lT4nQJ+g9srOzi5y3Lt3b5uammqttfbs2bO2d+/e9q233rLWWnvy5Enbu3dvu3DhQmuttUePHrW9e/e2S5YssdZae+jQIdu7d2+7bNkya621+/fvL/f9d+/ebQG7YcMGa621SUlJdsqUKdZaaw8fPuy6bsiQIa77tmzZ0p4+fdpaa+2PP/5orbV27Nixrjh//PFH27FjR3vixAm7du1a279/f2uttampqXb48OHWWmsfeOABO3jwYPvLL7/YrVu32iuuuMJaa+2qVavsQw89ZPPz8+0vv/xi+/fvb9etW1ck5oyMDBsbG+s6Loyhd+/eNj093fXvok2bNq73bdWqlevzLF261P7+97+31lp75swZGx4ebk+dOlUk1pEjR9o5c5z/PXz22Wf2lltusdZae+TIEZufn2+ttXb27Nn2ySefLPHZiiv+Z2yttUCGLSWvaoReyxXfau9Mh5u54JrBruPt27czZswYBg0axNTMi53z6cVp5F5rRURE0KtXLwCGDBnChg3ObqBr164lJiaGrl278vHHH7N161YAunXrxn333cf8+fNd0xerV68mJSWFqKgoHA4Hp0+fZs+ePZ7fsMAdd9zBBRdcQOfOnV0j3dWrV7N69Wq6d+/ONddcw/bt2/nmm2+K/Fz79u3ZtWsXjz32GCtXrqRxY8/Pn9z16dOHSy5x/rbbr18/1q5dy5kzZ/jwww+58cYbadCgQZHr4+PjWbRoEQALFy4kPj4egJycHG699Va6du3KlClTXP9OfElz6LWcp82xSxXRA5JGl3xd8+w1Qlpamuv7evXqFTlu2LBhkeMmTZoUOQ4LCyty3KKFp+XHJRljShyfPn2aRx99lIyMDCIiIpg4cSKnT58G4P3332f9+vUsX76cP/3pT2zZsgVrLUuWLCEyMrLIvcqakvjVr37l+t4WTOdYaxk7diwPP/xwqT/XrFkzvvrqK1atWsXrr7/O22+/zZw5c6hbty75+fkArlgLubcErl+/Pg6Hg1WrVrFo0SLXtJG7nj17snPnTg4dOsS7777Ls88+C8Bjjz3Gk08+SVxcHGlpaUycOLHUOCtLI3QpobDOvfBrweY9LFmyhNGjRxMXF8f06dNd1/br149Xt19yfuSu0XqtsmfPHjZtcg4CFixYwA033OBKiGFhYZw4ccI1x52fn8/evXu56aabeOmllzh27BgnTpzg1ltv5c9//rMrMf/zn/+sVCy33norc+bM4cSJEwDs27ePgwcPFrkmNzeX/Px8Bg0axAsvvMCXX34JOLe8++KLLwCKzMl7Eh8fT2pqKp988gl9+/Ytcd4Yw5133smTTz5Jp06daN68OQDHjh2jVStnAcK8efMq9RnLoxG6FFG8csarWvc2PSGpYCNgNQyrVSIjI5k5cybDhg2jc+fOPPLIIzRs2JCHHnqILl260KJFC3r06AE4H0gOGTKEY8eOYa1l5MiRNG3alHHjxjFq1Ci6detGfn4+7dq1Y8WKFRWO5Xe/+x3btm2jZ8+eAFx88cXMnz+fyy67zHXNvn37SEpKco3GCxfkjR49mnvuuYdZs2bRv38pf3fd3uf+++9n4MCBXHjhhR6viY+Pp0ePHsydO9f12sSJE7n77rtp1qwZN998M7t3767wZyyP6tClTMVr3cutbXdv71ucatp9KtB16OJ/qkMXn/JmZWpsbCw5OTnOdgPF69jdlTZqFxGfUEKXMrk/NC0crRd/cHq0xbXc0KMP4KznhaIP6IrwNB2jaRgRn9BDUfFa8RLHQmc63MxPbW8EnAsthg4dSl5eHg6Hg/nz5wNw6tQpHK98e36RUiE9RK2SQE2Ziv9V5s9WI3TxWml92t1H7EOHDgUgLy+vxHXH8hvQ5LYxUFCXC5T9EFUj9zLVr1+fw4cP07x58xLlgxLcrLUcPnyY+vXrV+jnlNDFJzy19C1eB+2xHK3r4JKvgdr4eqFwM+VDhw4FOhTxg/r16xMeHl6hn1FClyrztqXvsWPHSElJoUePHtx1113k5uYyePQ8EhMTSU5OLnqxHqCWq169erRr1y7QYUgN4lVCN8b0BaYDdYA3rbUpxc63BuYBTQuuGWOt/cDHsUoN5c2DU4BvP3mXU9nrXXXJ4Jyaadmypecbl1XP7k5TMyKAFwndGFMHmAn0AXKAdGPMMmtttttlzwJvW2tfM8Z0Bj4A2vohXqnhymrpe6bDzXT97R3cdZdz4UdYWBiffvqp54tLm4opTlMzIi7ejNCvA3Zaa3cBGGMWAgMB94RugcLyhybAv30ZpASP0h6cguc+MQcOHCAlJYW+ffvSt29f9u7dy/3338+zzz5LrDcLkDQ1I+LiTUJvBex1O84BYopdMxFYbYx5DLgIiPV0I2NMMpAM0Lp15bdNk+BVfDrGOQ2z2WNPDK8VTs1o6kVqOV/Vod8LzLXWhgO3AW8ZY0rc21o7y1obba2NvvTSS3301hIsPNWxn+lwM10f/h9XQo+IiCAtLc3V5rRcXQc72wmonl3Eq4S+D4hwOw4veM3dg8DbANbaTUB9IMwXAUro8LTDUmn7oQ4bNozhw4cDsGPHDhwOBxs3bgQgKysLh8NBenq6c0Se9L7nPu0itYw3Uy7pQEdjTDuciTwBSCx2zR7gFmCuMaYTzoSu4liptMTExBL9sT3JzMwEIMrfAYkEgXITurX2nDFmBLAKZ0niHGvtVmPMJJxbIS0DngJmG2OewPmAdKjVmmTxkucyx65cXN9ZMRMZGVlkkVKXLl1cx67eMQ9c5H2ZI2i+XUKSV3XoBTXlHxR7bbzb99lAL9+GJrVBaWWOXvVhB6ZNm+b85lwFNkVQqaOEKK0UlYDypj9MWaKiokhPT8fxdCkrTj1RqaOEKCV0qbFKW3FaXJcLcwkPD+f666/3/uYqdZQQpIQuNVJZK07dZe8/Di3DWFTQptcrhatQNfUiIUZb0ElQKxzBP97VkpKSwpQpU4iMjGTdunVMmDCBOXPm0L59e9asWcMLL7zAW2+9RUREQRVu4dSLtsSTIKIt6CTkZWVlcfz4ca+uXblyJQB9oWhljKZfJMhphC5BzX0T63I3sC7gKnWc+sD51aXawFqChEboErIK59q9LXMEWLhwofObFi3Oj8hV+SIhQHuKSlArbCdQWgsBT1q0aOEsdXQ4mDVrlh+jE6leGqFLrRUdHc1dd911/oWyVppqfl2CgBK6hAxv69YBBkb9B1On3n7+hbI21FB5owQJJXQJCd7WrcP5+fb6+zJISUnho48+okl0UukJW/PrEiSU0CUklLVTUnGFo/jc3FwaNmwIwKJFi5zn4uM9/9D3GyAjVaN0qdH0UFRqreTkZNLS0mjSpAmvvfYar732mucLC6djtIGG1HAaoUutVDjfXli7/sEHH5R+cXSSkrkEBY3QpdYp3Aove/9x3st0br7VsGFDli5disPhIC8vD4C5c+e6FiGJBAON0KXWKZxv97YixkUbaEgNp6X/Umu5tw0ASm0dMHv2bPjuUx7q4OWuimojIH6kpf8iHriXOpbVOqCwAuahP63x7sYqc5QAUUKXWsu91LGs6Zc1a7xM5CIBpoQu4oVXX30VgEcffdS7H1BbXgkAVbmIeGH58uUsX77cu4u7DnbOoYMzsavkUaqJRugiBcrqBXP/xDe8XomKexsBzadLNdIIXYTztemeFNarT58+nbi4ONfrU6dOZdCgQa7jlJQUEhIS/B6rSGk0Qheh7F4wFa5XL059YKSaqA5dpByFCX3Rwz29uv75558HYNy4cc5EvmIUtLlBdeniE2XVoWvKRcQLm3cfYcHmPV5du2PHDnbs2OE8iE5yJnORaqApF5FyDIxqxebdR3gvc59XD0bnz58PwPjx49m7dy+pN/o7QhEnJXSRciTGtHY18aq0ivSBqS6qjw85SugifjJp0iTnNxmpgQ3EE22rF5KU0EW8VLE9S52NvsaOHQvA5Mk17IFoTfttQXxCCV3EC5XZszQxpjWHDx/2V0giJSihi3ihMnuWAsyaNctfIYmUoLJFET8bPXo0o0ePDnQYJRUueJKQoYQu4geF8+3xb2ziq+8O8vPPPwc6pKK08XVI0pSLiI8V3zijc59HmOnlKtNqo42vQ5JG6CI+lhjTmkUP92TRwz1dDb9GjRrFqFGjXNcMHz68yDRMcnKyqyIGICkpifHjx7uOhwwZ4mopAJCQkEBKSorreNCgQUydOtV1HBcXx/Tp013H/fr1c/V0B4iNjWVWmncrXyV4KKGL1EI5OTkYE+goxNfUnEvEjyra2KtapfY/v6F1cVpFWmOpOZdIAFWksVe1ct9ZyZ12WQpaXj0UNcb0BaYDdYA3rbUpHq65B5gIWOAra22iD+MUCUoVbexVXRwOBwBpaWklT2oVadAqN6EbY+oAM4E+QA6QboxZZq3NdrumIzAW6GWt/dEYc5m/AhYJJj5p7OUHQ4cODXQI4gfejNCvA3Zaa3cBGGMWAgOBbLdrHgJmWmt/BLDWHvR1oCLiO0OHDiUvL4+8vDzq1asX6HDER7yZQ28F7HU7zil4zd2VwJXGmE+NMZ8VTNGUYIxJNsZkGGMyDh06VLmIRcQn+vTpQ58+fQIdhviQrxYW1QU6Ag4gHFhvjOlqrT3qfpG1dhYwC5xVLj56b5EaryKdGqtLZ8cdXN8hzPNJT/3bVflS43mT0PcBEW7H4QWvucsBNltr84DdxpivcSb4dJ9EKRLEKtKpsbpk7z9O55YxDBnioZyysC2AO/VPDwreJPR0oKMxph3ORJ4AFK9geRe4F0g1xoThnILZ5ctARYJVRTo1Vpf4NzZx7uxpxo8fz9VXX018fDzHjh1j4MCBjBw5kruKb2itypegUO4curX2HDACWAVsA9621m41xkwyxsQVXLYKOGyMyQbWAk9ba9UIWqQG+37zStavXx/oMMSHtFJUpBYqawVrbm4uAGFhbvPrhSP04iN3qXZaKSoiJZS2gnXw4MEMHlzKPHpqf+eX+qjXSGqfK1ILlbWC9amnnir5A+4PSvWAtMZSQhephcpawXr77beXfDE66XwC1wPSGktTLiJSxIEDBxg1ahQrV64EYO/evTgcDu2PGgSU0EWkiGXLlpGZmVnktbp169K+ffsARSTe0pSLiBSRnJxMcnKy6zgiIoI1a9YEMCLxlhK6SC1WWkuCgVGtijws3bXLuU7QNUr31BqgOLUKqHZK6CK1VGktCbL3HwcoktCHDRsGFPRP99QaoDhVwgSEErpILVVaSwJPI/bnnnvu/IF7xUtpVAkTEHooKiLl6t27Ny1atMDhcLBx48ZAhyOlUEIXEQkRSugi4pXIyEjS0tJo3LgxWVlZgQ5HPNAcuoiU4Kn6pbDyZcSIEUApG0xLQCmhi0gRnqpf3CtfpkyZUt0hiZeU0EWkCE/VL+6j9R49enh3o+K16qpL9zvNoYtIhWRmZpKQkOBqD5Ceno7D4XDNq2/cuBHHK98ya3uT8z90YAtsWRyIcGsVjdBFpEI+//xzDhw4UOY17br1pPeYMRAZ6XxBdenVQjsWiUi5ytrhyCva8chnytqxSCN0EfGKe+VL8V4vxa1btw5wLkiS6qOELiLlcq988dTrpbgJEyYAKm2sbkroIlIu98oXT71eipszZ46/QxIPlNBFxOc8bobhTctdUHljFSihi4jPFW6IERsb63zBm5a7oLa7VaSELiI+98ILL3D11VcTGxvLypUrSUmZR2JiYpGdkDxSeWOVKKGLiM8lJibSr18/13FUVBRxcXEBjKh2UEIXkQorbeu687py8b8tiRHQt29f+vbtW22x1WZK6CJSIaVtXeeueGnj8uXLAbj99tv9F5gooYtIxZS2dZ274qP3l19+GVBC9zcldBHxu8WL1ZirOiihi4jfhYWFBTqEWkHtc0XE75YuXcrSpUsDHUbIU0IXEb+bMWMG6enpACxatAiHw8GpU6cAmD9/Pg6Hg7y8vECGGBI05SIifpeYmEh8fHy5183dkAPAUC0UrRT1QxcRn4t/YxPZ+4/TuWVj12vltdwFcFzVHIC07Yf9Gl8wUz90EalWxWvVvWm5C5D2zG/8FlNtoIQuIj5XvFbdm5a7UnV6KCoiNcbsdXuYvW5PoMMIWhqhi0iNsejz/QA8FOA4gpUSuojUGGuejgl0CEHNq4RujOkLTAfqAG9aa1NKuW4QsBjoYa1VCYuIVJy3Oxt5qxbtgFRuQjfG1AFmAn2AHCDdGLPMWptd7LpGwOPAZn8EKiLBrfyWu1Dno3y6mQaMuc1Hb1rLdkDyZoR+HbDTWrsLwBizEBgIZBe77nngJeBpn0YoIkHPm5a7AMu/ymXtry5hzN/f980b17IdkLxJ6K2AvW7HOUCRiS5jzDVAhLX2fWNMqQndGJMMJAO0bl12PaqIhA5vWu4CxDOjGqIJXVUuWzTGXAD8D/BUeddaa2dZa6OttdGXXnppVd9aRETceDNC3wdEuB2HF7xWqBHQBUgzxgC0AJYZY+L0YFREKuLrfyxyfvNwz8AGEqS8SejpQEdjTDuciTwBSCw8aa09BriaHRtj0oDRSuYiUlEHd3wR6BCCWrkJ3Vp7zhgzAliFs2xxjrV2qzFmEpBhrV3m7yBFpHa44dH/DnQIQc2rOnRr7QfAB8VeG1/KtY6qhyUiIhWllaIiUmNsX73A+Y3m0CtFzblEpMY4vDuLw7uzAEhJSSEhIcF17vnnn2fIkCGu4/Hjx5OUdH7B0NixY0lOTnYdjx49muFvZVVD1DWHRugiUmP0evjFQIcQ1LRjkYjUGIWtARb5asoltT98vwEGTAuZ5f9l7VikKRcRCVmjVp9l1MrTsGVxoEOpFkroIhK6Lr8aGv060FFUG82hi8XBuoEAAAfPSURBVEjImjZtGqR+E+gwqo1G6CIiIUIJXURC1vDhw2tV6aKmXEQkZDVo0AAurBPoMKqNErqIhKypU6dC6rZAh1FtlNBFpEbxZqu6iphx5jSXN6rvs/vVZJpDF5EaY2BUKzq3bOyz+30063lGvLXVZ/er6TRCF5Eaw9ut6rzV6Z0wmvxce9Jc7fmkIlLrdLvzEZ44vCvQYVQbJXQRCWnPLvmGppwglf5Vu1HXwTW+H4wSuoiEtLzmVxJR/0DVbnJgi/OfSugiIoFz0aD/ZlJVuzemVnF0X01U5SIiEiKU0EUkpH02Z2KRnY5CmaZcRCSkNbq8DZGREYEOo1oooYtISLu6fxLjasmm05pyEREJEUroIhLSVkx7hp594gIdRrVQQheRkDUwqhUXXt6ekxe3AmDQoEHODowF4uLimD59uuu4X79+vPrqq67j2NhYZs+eXX0BV5Hm0EUkZCXGtOa9ex8OdBjVxlhrA/LG0dHRNiMjIyDvLSK1R2Er3kVVeDDquKo5AGnbD/skpqowxnxhrY32dE4jdBGRcgztFR7oELyihC4iUo6hNyihi4iEhLxz+fBDFvVqeE8XVbmIiJSjz+u76TP/VKDDKJdG6CIi5fjDE886v6kJPWGGmVJPKaGLiJQjWJp7KaGLSMjL3n/cVb5YGefOnmZAt1+T1DvSh1H5nhK6iIS0gVGtqnyPlS8/zicX1iXp6y99EJH/KKGLSEhLjGlNYkzrKt2jZ8bdPorGv5TQRUTK0To6NtAheEUJXUSkHGd/PhHoELyiOnQRkXJ8+uozrJ/2OAC5ubk4HA6WL18OwIEDB3A4HKxcuRKAvXv34nA4WLNmDQC7du3C4XCwbt06AHbs2IHD4WDjxo0AZGVl4XA4SE9PByAzMxOHw0FmZiYA6enpOBwOsrKyyo3Tq4RujOlrjNlhjNlpjBnj4fyTxphsY8y/jDH/MMa08ea+IiLBoPV1fbiq7/2BDqNc5XZbNMbUAb4G+gA5QDpwr7U22+2am4DN1tpTxphHAIe1Nr6s+6rboogEC190bPSVsrotejNCvw7Yaa3dZa09CywEBrpfYK1da60tXBf7GRAcnWxEREKINwm9FbDX7Tin4LXSPAh86OmEMSbZGJNhjMk4dOiQ91GKiEi5fPpQ1BgzBIgGpng6b62dZa2NttZGX3rppb58axGRWs+bssV9QITbcXjBa0UYY2KB/wJ6W2vP+CY8ERHxljcj9HSgozGmnTHmQiABWOZ+gTGmO/AGEGetPej7MEVEpDzlJnRr7TlgBLAK2Aa8ba3daoyZZIyJK7hsCnAx8HdjTKYxZlkptxMRET/xaqWotfYD4INir413+z441sWKiFRSVTs2Vgct/RcRKYcvOjZWh3IXFvmLFhaJiFRcVRcWiYhIEFBCFxEJEUroIiIhQgldRCREKKGLiIQIJXQRkRChhC4iEiKU0EVEQkTAFhYZYw4B3wfkzasmDMgNdBB+EsqfDfT5glkofzao2OdrY6312H88YAk9WBljMkpbpRXsQvmzgT5fMAvlzwa++3yachERCRFK6CIiIUIJveJmBToAPwrlzwb6fMEslD8b+OjzaQ5dRCREaIQuIhIilNBFREKEEroXjDERxpi1xphsY8xWY8zjgY7JH4wxdYwx/zTGrAh0LL5mjGlqjFlsjNlujNlmjOkZ6Jh8xRjzRMHfyyxjzN+MMfUDHVNVGGPmGGMOGmOy3F67xBjzkTHmm4J/NgtkjFVRyuebUvB381/GmHeMMU0rc28ldO+cA56y1nYGfgMMN8Z0DnBM/vA4zo3AQ9F0YKW19irgPwiRz2mMaQWMBKKttV2AOkBCYKOqsrlA32KvjQH+Ya3tCPyj4DhYzaXk5/sI6GKt7QZ8DYytzI2V0L1grd1vrf2y4PufcCaD4Nhk0EvGmHCgP/BmoGPxNWNME+BG4C8A1tqz1tqjgY3Kp+oCDYwxdYGGwL8DHE+VWGvXA0eKvTwQmFfw/TzgjmoNyoc8fT5r7Wpr7bmCw8+A8MrcWwm9gowxbYHuwObARuJz04D/BPIDHYgftAMOAakFU0pvGmMuCnRQvmCt3QdMBfYA+4Fj1trVgY3KLy631u4v+P4AcHkgg/GzYcCHlflBJfQKMMZcDCwBRllrjwc6Hl8xxgwADlprvwh0LH5SF7gGeM1a2x04SXD/yu5SMJc8EOf/tH4NXGSMGRLYqPzLOmutQ7Le2hjzXzineP+vMj+vhO4lY0w9nMn8/6y1SwMdj4/1AuKMMd8BC4GbjTHzAxuST+UAOdbawt+qFuNM8KEgFthtrT1krc0DlgLXBzgmf/jBGNMSoOCfBwMcj88ZY4YCA4D7bCUXCCmhe8EYY3DOv26z1v5PoOPxNWvtWGttuLW2Lc4Hah9ba0NmlGetPQDsNcZEFrx0C5AdwJB8aQ/wG2NMw4K/p7cQIg98i1kGPFDw/QPAewGMxeeMMX1xTnnGWWtPVfY+Suje6QXcj3PkmlnwdVugg5IKeQz4P2PMv4Ao4MUAx+MTBb91LAa+BLbg/G86qJfJG2P+BmwCIo0xOcaYB4EUoI8x5hucv5WkBDLGqijl870CNAI+Ksgvr1fq3lr6LyISGjRCFxEJEUroIiIhQgldRCREKKGLiIQIJXQRkRChhC4iEiKU0EVEQsT/B7YjD2mqmcoIAAAAAElFTkSuQmCC\n",
      "text/plain": [
       "<Figure size 432x288 with 1 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "cph.plot_covariate_groups('trt', values=[0, 1]);"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Notice how the group without treatment has a lower survival rate at all times (the x-axis is time) compared to the treatment group."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a name='6'></a>\n",
    "## 6. Hazard Ratio\n",
    "\n",
    "Recall from the lecture videos that the Hazard Ratio between two patients was the likelihood of one patient (e.g smoker) being more at risk than the other (e.g non-smoker).\n",
    "$$\n",
    "\\frac{\\lambda_{smoker}(t)}{\\lambda_{nonsmoker}(t)} = e^{\\theta (X_{smoker} - X_{nonsmoker})^T}\n",
    "$$\n",
    "\n",
    "Where\n",
    "\n",
    "$$\n",
    "\\lambda_{smoker}(t) = \\lambda_0(t)e^{\\theta X_{smoker}^T}\n",
    "$$\n",
    "and\n",
    "$$\n",
    "\\lambda_{nonsmoker}(t) = \\lambda_0(t)e^{\\theta X_{nonsmoker}^T} \\\\\n",
    "$$"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a name='ex-2'></a>\n",
    "### Exercise 2 - hazard_ratio\n",
    "In the cell below, write a function to compute the hazard ratio between two individuals given the cox model's coefficients."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<details>    \n",
    "<summary>\n",
    "    <font size=\"3\" color=\"darkgreen\"><b>Hints</b></font>\n",
    "</summary>\n",
    "<p>\n",
    "<ul>\n",
    "    <li>use numpy.dot</li>\n",
    "    <li>use nump.exp</li>\n",
    "</ul>\n",
    "</p>\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "WbBmxbeDA3k1"
   },
   "outputs": [],
   "source": [
    "# UNQ_C2 (UNIQUE CELL IDENTIFIER, DO NOT EDIT)\n",
    "def hazard_ratio(case_1, case_2, cox_params):\n",
    "    '''\n",
    "    Return the hazard ratio of case_1 : case_2 using\n",
    "    the coefficients of the cox model.\n",
    "    \n",
    "    Args:\n",
    "        case_1 (np.array): (1 x d) array of covariates\n",
    "        case_2 (np.array): (1 x d) array of covariates\n",
    "        model (np.array): (1 x d) array of cox model coefficients\n",
    "    Returns:\n",
    "        hazard_ratio (float): hazard ratio of case_1 : case_2\n",
    "    '''\n",
    "    \n",
    "    ### START CODE HERE (REPLACE INSTANCES OF 'None' with your code) ###\n",
    "    \n",
    "    hr = np.exp(cox_params.dot((case_1 - case_2).T))\n",
    "    \n",
    "    ### END CODE HERE ###\n",
    "    \n",
    "    return hr"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "zbDQUxE6CcA3"
   },
   "source": [
    "Now, evaluate it on the following pair of indivduals: `i = 1` and `j = 5`. Given your implementation of `hazard_ratio` is correct, you will **only** pass the test when `i = 1` and `j = 5`."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "7flsvTRXCgqO"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "15.029017732492221 \n",
      "\n",
      "\u001b[92m All tests passed.\n"
     ]
    }
   ],
   "source": [
    "# test cell ex2\n",
    "\n",
    "# Set i = 1 and j = 5 to pass the test\n",
    "i = 1\n",
    "j = 5\n",
    "\n",
    "# DO NOT modify anything below\n",
    "hazard_ratio_test(hazard_ratio, i, j, one_hot_train, cph)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Expected Output:\n",
    "```\n",
    "15.029017732492221 \n",
    "\n",
    " All tests passed.\n",
    "```"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "**Question:** \n",
    "\n",
    "Is `case_1` or `case_2` at greater risk? "
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<details>    \n",
    "<summary>\n",
    "    <font size=\"3\" color=\"darkgreen\"><b>Check your answer!</b></font>\n",
    "</summary>\n",
    "<p>\n",
    "<ul>\n",
    "<ul>\n",
    "    Important! The following answer only applies if you picked i = 1 and j = 5\n",
    "    <li>You should see that `case_1` is at higher risk.</li>\n",
    "    <li>The hazard ratio of case 1 / case 2 is greater than 1, so case 1 had a higher hazard relative to case 2</li>\n",
    "</ul>\n",
    "</p>"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Inspect different pairs, and see if you can figure out which patient is more at risk."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 374
    },
    "colab_type": "code",
    "id": "g2PZ3sGvCs0K",
    "outputId": "59336868-d421-4645-d88e-76a8a8cffc9f"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Case 1\n",
      "\n",
      " trt          0.000000\n",
      "age          0.563645\n",
      "sex          0.000000\n",
      "ascites      0.000000\n",
      "hepato       1.000000\n",
      "spiders      1.000000\n",
      "bili        -0.405645\n",
      "chol        -0.210436\n",
      "albumin      1.514297\n",
      "copper      -0.481961\n",
      "alk.phos     2.173526\n",
      "ast         -0.144699\n",
      "trig        -0.531125\n",
      "platelet    -0.450972\n",
      "protime     -0.139881\n",
      "edema_0.5    0.000000\n",
      "edema_1.0    0.000000\n",
      "stage_2.0    0.000000\n",
      "stage_3.0    1.000000\n",
      "stage_4.0    0.000000\n",
      "Name: 1, dtype: float64 \n",
      "\n",
      "Case 2\n",
      "\n",
      " trt          0.000000\n",
      "age          0.463447\n",
      "sex          0.000000\n",
      "ascites      0.000000\n",
      "hepato       1.000000\n",
      "spiders      0.000000\n",
      "bili        -0.489581\n",
      "chol        -0.309875\n",
      "albumin     -1.232371\n",
      "copper      -0.504348\n",
      "alk.phos     2.870427\n",
      "ast         -0.936261\n",
      "trig        -0.150229\n",
      "platelet     3.190823\n",
      "protime     -0.139881\n",
      "edema_0.5    0.000000\n",
      "edema_1.0    0.000000\n",
      "stage_2.0    0.000000\n",
      "stage_3.0    0.000000\n",
      "stage_4.0    1.000000\n",
      "Name: 38, dtype: float64 \n",
      "\n",
      "Hazard Ratio: 0.1780450006997129\n"
     ]
    }
   ],
   "source": [
    "i = 4\n",
    "case_1 = one_hot_train.iloc[i, :].drop(['time', 'status'])\n",
    "\n",
    "j = 7\n",
    "case_2 = one_hot_train.iloc[j, :].drop(['time', 'status'])\n",
    "\n",
    "print(\"Case 1\\n\\n\", case_1, \"\\n\")\n",
    "print(\"Case 2\\n\\n\", case_2, \"\\n\")\n",
    "print(\"Hazard Ratio:\", hazard_ratio(case_1.values, case_2.values, cph.params_.values))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<details>    \n",
    "<summary>\n",
    "    <font size=\"3\" color=\"darkgreen\"><b>Check your answer!</b></font>\n",
    "</summary>\n",
    "<p>\n",
    "<ul>\n",
    "<ul>\n",
    "    Important! The following answer only applies if you picked i = 4 and j = 7\n",
    "    <li>You should see that `case_2` is at higher risk.</li>\n",
    "    <li>The hazard ratio of case 1 / case 2 is less than 1, so case 2 had a higher hazard relative to case 1</li>\n",
    "</ul>\n",
    "</p>"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "KUa6r-KOyySp"
   },
   "source": [
    "<a name='7'></a>\n",
    "## 7. Harrell's C-index"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "woQAtSmRXrgr"
   },
   "source": [
    "To evaluate how good our model is performing, we will write our own version of the C-index. Similar to the week 1 case, C-index in the survival context is the probability that, given a randomly selected pair of individuals, the one who died sooner has a higher risk score. \n",
    "\n",
    "However, we need to take into account censoring. Imagine a pair of patients, $A$ and $B$. \n",
    "\n",
    "#### Scenario 1\n",
    "- A was censored at time $t_A$ \n",
    "- B died at $t_B$\n",
    "- $t_A < t_B$. \n",
    "\n",
    "Because of censoring, we can't say whether $A$ or $B$ should have a higher risk score. \n",
    "\n",
    "#### Scenario 2\n",
    "Now imagine that $t_A > t_B$.\n",
    "\n",
    "- A was censored at time $t_A$ \n",
    "- B died at $t_B$\n",
    "- $t_A > t_B$\n",
    "\n",
    "Now we can definitively say that $B$ should have a higher risk score than $A$, since we know for a fact that $A$ lived longer. \n",
    "\n",
    "Therefore, when we compute our C-index\n",
    "- We should only consider pairs where at most one person is censored\n",
    "- If they are censored, then their censored time should occur *after* the other person's time of death. \n",
    "\n",
    "The metric we get if we use this rule is called **Harrel's C-index**.\n",
    "\n",
    "Note that in this case, being censored at time $t$ means that the true death time was some time AFTER time $t$ and not at $t$. \n",
    "- Therefore if $t_A = t_B$ and A was censored:\n",
    "    - Then $A$ actually lived longer than $B$. \n",
    "    - This will effect how you deal with ties in the exercise below!\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a name='ex-3'></a>\n",
    "### Exercise 3 - harrell_c\n",
    "Fill in the function below to compute Harrel's C-index."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<details>    \n",
    "<summary>\n",
    "    <font size=\"3\" color=\"darkgreen\"><b>Hints</b></font>\n",
    "</summary>\n",
    "<p>\n",
    "<ul>\n",
    "    <li>If you get a division by zero error, consider checking how you count when a pair is permissible (in the case where one patient is censored and the other is not censored).</li>\n",
    "</ul>\n",
    "</p>"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 23,
   "metadata": {},
   "outputs": [],
   "source": [
    "# UNQ_C3 (UNIQUE CELL IDENTIFIER, DO NOT EDIT)\n",
    "def harrell_c(y_true, scores, event):\n",
    "    '''\n",
    "    Compute Harrel C-index given true event/censoring times,\n",
    "    model output, and event indicators.\n",
    "    \n",
    "    Args:\n",
    "        y_true (array): array of true event times\n",
    "        scores (array): model risk scores\n",
    "        event (array): indicator, 1 if event occurred at that index, 0 for censorship\n",
    "    Returns:\n",
    "        result (float): C-index metric\n",
    "    '''\n",
    "    \n",
    "    n = len(y_true)\n",
    "    assert (len(scores) == n and len(event) == n)\n",
    "    \n",
    "    concordant = 0.0\n",
    "    permissible = 0.0\n",
    "    ties = 0.0\n",
    "    \n",
    "    result = 0.0\n",
    "    \n",
    "    ### START CODE HERE (REPLACE INSTANCES OF 'None' and 'pass' with your code) ###\n",
    "    \n",
    "    # use double for loop to go through cases\n",
    "    for i in range(n):\n",
    "        # set lower bound on j to avoid double counting\n",
    "        for j in range(i + 1, n):\n",
    "            \n",
    "            # check if at most one is censored\n",
    "            if event[i] == 1 or event[j] == 1:\n",
    "                pass\n",
    "            \n",
    "                # check if neither are censored\n",
    "                if event[i] == 1 and event[j] == 1:\n",
    "                    permissible += 1\n",
    "                    \n",
    "                    # check if scores are tied\n",
    "                    if scores[i] == scores[j]:\n",
    "                        ties += 1\n",
    "                    \n",
    "                    # check for concordant\n",
    "                    elif y_true[i] < y_true[j] and scores[i] > scores[j]:\n",
    "                        concordant += 1.0\n",
    "                    elif y_true[i] > y_true[j] and scores[i] < scores[j]:\n",
    "                        concordant += 1.0\n",
    "                \n",
    "                # check if one is censored\n",
    "                elif event[i] != event[j]:\n",
    "                    \n",
    "                    # get censored index\n",
    "                    censored = j\n",
    "                    uncensored = i\n",
    "                    \n",
    "                    if event[i] == 0:\n",
    "                        censored = i\n",
    "                        uncensored = j\n",
    "                        \n",
    "                    # check if permissible\n",
    "                    # Note: in this case, we are assuming that censored at a time\n",
    "                    # means that you did NOT die at that time. That is, if you\n",
    "                    # live until time 30 and have event = 0, then you lived THROUGH\n",
    "                    # time 30.\n",
    "                    if y_true[uncensored] <= y_true[censored]:\n",
    "                        permissible += 1\n",
    "                        \n",
    "                        # check if scores are tied\n",
    "                        if scores[uncensored] == scores[censored]:\n",
    "                            # update ties \n",
    "                            ties += 1.0\n",
    "                            \n",
    "                        # check if scores are concordant \n",
    "                        if scores[uncensored] > scores[censored]:\n",
    "                            concordant += 1.0\n",
    "    \n",
    "    # set result to c-index computed from number of concordant pairs,\n",
    "    # number of ties, and number of permissible pairs (REPLACE 0 with your code)  \n",
    "    result = (concordant + 0.5 * ties) / permissible\n",
    "    \n",
    "    ### END CODE HERE ###\n",
    "    \n",
    "    return result   "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Test Case 1\n",
      "\n",
      "y_true:  [30, 12, 84, 9]\n",
      "scores:  [0.5, 0.9, 0.1, 1.0]\n",
      "event:   [1, 1, 1, 1]\n",
      "Output:  1.0\n",
      "\n",
      "Test Case 2\n",
      "\n",
      "y_true:  [30, 12, 84, 9]\n",
      "scores:  [0.9, 0.5, 1.0, 0.1]\n",
      "event:   [1, 1, 1, 1]\n",
      "Output:  0.0\n",
      "\n",
      "Test Case 3\n",
      "\n",
      "y_true:  [30, 12, 84, 9]\n",
      "scores:  [0.5, 0.9, 0.1, 1.0]\n",
      "event:   [1, 0, 1, 1]\n",
      "Output:  1.0\n",
      "\n",
      "Test Case 4\n",
      "\n",
      "y_true:  [30, 30, 20, 20]\n",
      "scores:  [10, 5, 15, 20]\n",
      "event:   [1, 0, 1, 0]\n",
      "Output:  0.75\n",
      "\n",
      "Test Case 5\n",
      "\n",
      "y_true:  [20, 20, 30, 30, 30]\n",
      "scores:  [20, 15, 5, 10, 15]\n",
      "event:   [0, 1, 0, 1, 0]\n",
      "Output:  0.5833333333333334\n",
      "\n",
      "Tes Case 6\n",
      "\n",
      "y_true:  [10, 10]\n",
      "scores:  [4, 5]\n",
      "event:   [0, 1]\n",
      "Output:  1.0 \n",
      "\n",
      "\u001b[92m All tests passed.\n"
     ]
    }
   ],
   "source": [
    "# test cell ex1 - do not modify this test cell\n",
    "harrell_c_test(harrell_c)    "
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "#### Expected Output:\n",
    "```\n",
    "Test Case 1\n",
    "Output:  1.0\n",
    "\n",
    "Test Case 2\n",
    "Output:  0.0\n",
    "\n",
    "Test Case 3\n",
    "Output:  1.0\n",
    "\n",
    "Test Case 4\n",
    "Output:  0.75\n",
    "\n",
    "Test Case 5\n",
    "Output:  0.5833333333333334\n",
    "\n",
    "Tes Case 6\n",
    "Output:  1.0 \n",
    "\n",
    " All tests passed.\n",
    "```"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "CtQVe4pAn8ic"
   },
   "source": [
    "Now use the Harrell's C-index function to evaluate the cox model on our data sets."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 34
    },
    "colab_type": "code",
    "id": "8nzHc_Qbn7dM",
    "outputId": "bc2f960d-16e5-46b2-a41f-695892c311c7"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Train: 0.8265139116202946\n",
      "Val: 0.8544776119402985\n",
      "Test: 0.8478543563068921\n"
     ]
    }
   ],
   "source": [
    "# Train\n",
    "scores = cph.predict_partial_hazard(one_hot_train)\n",
    "cox_train_scores = harrell_c(one_hot_train['time'].values, scores.values, one_hot_train['status'].values)\n",
    "# Validation\n",
    "scores = cph.predict_partial_hazard(one_hot_val)\n",
    "cox_val_scores = harrell_c(one_hot_val['time'].values, scores.values, one_hot_val['status'].values)\n",
    "# Test\n",
    "scores = cph.predict_partial_hazard(one_hot_test)\n",
    "cox_test_scores = harrell_c(one_hot_test['time'].values, scores.values, one_hot_test['status'].values)\n",
    "\n",
    "print(\"Train:\", cox_train_scores)\n",
    "print(\"Val:\", cox_val_scores)\n",
    "print(\"Test:\", cox_test_scores)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "What do these values tell us ?"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "AuNjR_wNkpWz"
   },
   "source": [
    "<a name='8'></a>\n",
    "## 8. Random Survival Forests\n",
    "\n",
    "This performed well, but you have a hunch you can squeeze out better performance by using a machine learning approach. You decide to use a Random Survival Forest. To do this, you can use the `RandomForestSRC` package in R. To call R function from Python, we'll use the `r2py` package. Run the following cell to import the necessary requirements. \n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 530
    },
    "colab_type": "code",
    "id": "ZgSy-Dj6kquK",
    "outputId": "4aa5d2fa-30f4-4328-ae29-a2ff05223e22"
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "R[write to console]: Loading required package: ggplot2\n",
      "\n"
     ]
    }
   ],
   "source": [
    "%load_ext rpy2.ipython\n",
    "%R require(ggplot2)\n",
    "\n",
    "from rpy2.robjects.packages import importr\n",
    "# import R's \"base\" package\n",
    "base = importr('base')\n",
    "\n",
    "# import R's \"utils\" package\n",
    "utils = importr('utils')\n",
    "\n",
    "# import rpy2's package module\n",
    "import rpy2.robjects.packages as rpackages\n",
    "\n",
    "forest = rpackages.importr('randomForestSRC', lib_loc='R')\n",
    "\n",
    "from rpy2 import robjects as ro\n",
    "R = ro.r\n",
    "\n",
    "from rpy2.robjects import pandas2ri\n",
    "pandas2ri.activate()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "LXBgqQBfuMA5"
   },
   "source": [
    "Instead of encoding our categories as binary features, we can use the original dataframe since trees deal well with raw categorical data (can you think why this might be?).\n",
    "\n",
    "Run the code cell below to build your forest."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "B-pio4o4mdVJ"
   },
   "outputs": [],
   "source": [
    "model = forest.rfsrc(ro.Formula('Surv(time, status) ~ .'), data=df_train, ntree=300, nodedepth=5, seed=-1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 289
    },
    "colab_type": "code",
    "id": "zZfcUvJ3nL04",
    "outputId": "27d00bd8-ea33-4c1b-f5d7-ca2c73ead721"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "                         Sample size: 154\n",
      "                    Number of deaths: 64\n",
      "                     Number of trees: 300\n",
      "           Forest terminal node size: 15\n",
      "       Average no. of terminal nodes: 6.54\n",
      "No. of variables tried at each split: 5\n",
      "              Total no. of variables: 17\n",
      "       Resampling used to grow trees: swor\n",
      "    Resample size used to grow trees: 97\n",
      "                            Analysis: RSF\n",
      "                              Family: surv\n",
      "                      Splitting rule: logrank *random*\n",
      "       Number of random split points: 10\n",
      "                          Error rate: 19.07%\n",
      "\n",
      "\n"
     ]
    }
   ],
   "source": [
    "print(model)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "9Mwzm55H-QKV"
   },
   "source": [
    "Finally, let's evaluate on our validation and test sets, and compare it with our Cox model."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 34
    },
    "colab_type": "code",
    "id": "vfl4LbGfpbKp",
    "outputId": "13f8b560-e171-41e9-f6dc-cf0468a3f786"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Cox Model Validation Score: 0.8544776119402985\n",
      "Survival Forest Validation Score: 0.8296019900497512\n"
     ]
    }
   ],
   "source": [
    "result = R.predict(model, newdata=df_val)\n",
    "scores = np.array(result.rx('predicted')[0])\n",
    "\n",
    "print(\"Cox Model Validation Score:\", cox_val_scores)\n",
    "print(\"Survival Forest Validation Score:\", harrell_c(df_val['time'].values, scores, df_val['status'].values))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 34
    },
    "colab_type": "code",
    "id": "uhqSQJhrplSG",
    "outputId": "752c266e-0234-45c5-d53f-554e2ff17a5a"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Cox Model Test Score: 0.8478543563068921\n",
      "Survival Forest Validation Score: 0.8621586475942783\n"
     ]
    }
   ],
   "source": [
    "result = R.predict(model, newdata=df_test)\n",
    "scores = np.array(result.rx('predicted')[0])\n",
    "\n",
    "print(\"Cox Model Test Score:\", cox_test_scores)\n",
    "print(\"Survival Forest Validation Score:\", harrell_c(df_test['time'].values, scores, df_test['status'].values))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "Gp_SgUXreAWn"
   },
   "source": [
    "Your random forest model should be outperforming the Cox model slightly. Let's dig deeper to see how they differ."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "ZtPMPaSli8GB"
   },
   "source": [
    "<a name='9'></a>\n",
    "## 9. Permutation Method for Interpretation\n",
    "\n",
    "We'll dig a bit deeper into interpretation methods for forests a bit later, but for now just know that random surival forests come with their own built in variable importance feature. The method is referred to as VIMP, and for the purpose of this section you should just know that higher absolute value of the VIMP means that the variable generally has a larger effect on the model outcome.\n",
    "\n",
    "Run the next cell to compute and plot VIMP for the random survival forest."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 281
    },
    "colab_type": "code",
    "id": "u7M4_N_d-YJu",
    "outputId": "7e1830cb-4b67-444f-8ba5-d49d3ff2f172"
   },
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAY4AAAEICAYAAABI7RO5AAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4xLjIsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy8li6FKAAAgAElEQVR4nO3deZxcZZn28d+VgISwJEBQAcEWBNmJ2KAsKiI6CI4ioCggImoGQXjxHVQcmREcF5R3xo1xiQyyBDfCgA5oQGVRUAwJCVmIoJIogiJhCUtYQnK9f5ynpSiqk67uqq7q5Pp+Pv3JqXOe85y7Kknf9ZznnPvINhEREQM1qtMBRETEyJLEERERTUniiIiIpiRxREREU5I4IiKiKUkcERHRlCSOWONJ+pykUwbY9jpJ72/x8c+QNKWVfbaDpEWSDmhBPydJ+nwrYorOSOKIjpA0TdKnGqx/q6S/SlpL0vmSPl3W90iypFl17SdIekrSopp1iyQ9LulRSfeWftbvJ45NgWOAb7b0DbbJSEkyq/At4ChJz+90IDE4SRzRKRcAR0tS3fp3Axfbfrqf/cZK2rnm9ZHAwgbt/tH2+sDuQC9wej/9HQv82PbjA448hsT2E8BPqBJ2jEBJHNEplwObAK/uWyFpI+DNwIUr2e8i4D01r49ZWXvbd1P9ktq5nyZvAq6vjUHSFZLuk/RgWX5R3T7bSJou6WFJP5S0cdl3jKQpku6X9JCkmyW9oGzbXNKPJD0g6feSPtAoGEn7Sfpz3bpFkg6QdCDwL8ARZTR1a9k+TtJ/S/qLpLslfVrS6AZ9b15GYhvXrHu5pMWS1pa0jaRrSvyLJV0saXw/cf59NNgo7nKsS8vnuFDSyXVdXAcc3Kjv6H5JHNER5Rv+D3j2t853AL+1fetKdp0CvFPSaEk7AusDv+mvsaQtgYOAWf002QW4veb1KODbwIuBrYDHgXPq9jkGOA7YDHga+EpZ/x5gHLAlVVI8vuwP8D3gz8DmwOHAZyXtv5L3+Ry2pwGfBb5ve33bu5VN55c4Xgq8HHgj8Jx5GNv3AL8GDqtZfSQw1fYyQMDnSow7lPdxRjMxAkgaBfwvcCuwBfB64BRJ/1DTbAGwW4PdYwRI4ohOugA4XNKY8vqYsm5l/kz1i/6A0v6iftpdLukh4AaqEcVn+2k3Hnik74Xt+21fanup7UeAzwCvrdvnItvzbD8G/CvwjvINfxlVwnip7eW2Z9p+uCSvfYCP2X7C9mzgXFpwqqaMaA4CTrH9mO2/AV8E3tnPLt8B3lX2VWn3nfLef2/7p7aftH0f8J8N3vtA7AFsavtTtp+yfSfVvEZtTI9QJdkYgdbqdACx5rJ9g6TFwCGSbgb2BA4dwK4XUs1N7E11qmu7Bm0Osf2zAfT1ILBB3wtJY6l+8R4IbFRWbyBptO3l5fVdNfv/EVgbmECVxLYEvldO8UwBPkH1Df6Bkohq9+sdQHyr8uJy/L/UTBeNqoux1qXAVyVtRvW5rQB+CX9PQl+m+kw3KP08OMiYNi+Ju8/ovuMUGwBLBtF3dIGMOKLTLqT65n00cJXtewewz6VU58fvtP2nIR5/Ds9OPP8MvAx4pe0NgdeU9bWT+FvWLG9FNdJYbHuZ7TNt70iV1N5M9d7uATaWtEHdfnc3iOcxYGzfizKS2bRme30567uAJ4EJtseXnw1t79Tozdp+ELgaOILqNNX3/EyJ7M+W/ncp7/3ouvfdb5zAC+tiWlgTz3jbG9g+qKbNDlSnsmIESuKITruQ6rTTB1j1aSoAyimi/WlwHn8QfsyzT8dsQDUv8VCZRP5kg32OlrRjGZ18imqOYLmk10napfyyf5gqoaywfRfwK+BzZQJ9V+B9VCOSencAYyQdLGltqqvB1qnZfi/QU+YRsP0XqkTwH5I2lDSqTHKv7BTTd6gS2uFlufa9PwoskbQF8JGV9DEbOEjSxpJeCNTeBzMdeETSxyStW+ajdpa0R02b11JdtBAjUBJHdJTtRVS/VNcDftTEfjNs/6EFIVxI9Qtw3fL6S8C6wGLgJmBag30uopqQ/iswBui7YuiFwFSqpLGAam6lbw7mXUAP1ejjMuCTjU6l2V4CnEA1B3I31Tf72qusLil/3i/plrJ8DPA84DaqU0tTqSbu+/MjYFvgr3UXIpxJdfnyEuBK4H9W0sdFVCOGRVSJ6/s172E51WhrItWl0ovL+xkH1dVnVPMyA/qiEN1HeZBTrOkkfRb4m+0vdTqWNYGkk4AtbX+007HE4CRxREREU3KqKiIimpLEERERTUniiIiIpqz2NwBOmDDBPT09nQ4jImJEmTlz5mLbmzbattonjp6eHmbMmNHpMCIiRhRJf+xvW05VRUREU5I4IiKiKUkcERHRlCSOiIhoShJHREQ0JYkjIiKaksQRERFNSeKIiIimrPY3AM69ewk9p125ynaLzjp4GKKJiBj5MuKIiIimDGvikHRKedxmRESMUMM94jiFZz/gPiIiRpi2JQ5J60m6UtKtkuZJ+iSwOXCtpGtLm69LmiFpvqQza/Y9SNJvJc2U9BVJV9T0eZ6k6ZJmSXpru+KPiIjG2jk5fiBwj+2DASSNA94LvM724tLmE7YfkDQa+LmkXYE7gG8Cr7G9UNJ3a/r8BHCN7eMkjQemS/qZ7cdqDyxpEjAJYPSGDasCR0TEILXzVNVc4A2SPi/p1baXNGjzDkm3ALOAnYAdge2BO20vLG1qE8cbgdMkzQauA8YAW9V3anuy7V7bvaPHjmvdO4qIiPaNOGzfIWl34CDg05J+Xrtd0kuAU4E9bD8o6XyqRLAyAg6zfXs7Yo6IiFVr5xzH5sBS21OAs4HdgUeADUqTDYHHgCWSXgC8qay/HdhaUk95fURNt1cBJ0lSOcbL2xV/REQ01s45jl2AsyWtAJYBHwT2AqZJusf26yTNAn4L3AXcCGD7cUknlHaPATfX9PnvwJeAOZJGAQuBN7fxPURERB3Z7nQMzyFpfduPlpHFfwG/s/3FwfTV29vrPDo2IqI5kmba7m20rVvvHP9AmQCfD4yjusoqIiK6QFeOOFppnc229Wbv+dKwHjN1ryJipBuJI46IiOhSw12r6thytVXf63Ml7TicMURExNC0PHGUu8D7cyxV2REAbL/f9m2tjiEiItqnqcQhqafUkLpY0gJJUyWNlbSo3CF+C/B2SRMl3SRpjqTLJG0k6XCgF7hY0mxJ60q6TlJv6ftRSWeXulU/k7Rn2X6npLeUNqNLm5tL3//U8k8kIiJWajAjjpcBX7O9A/AwcEJZf7/t3W1/D7gQ+JjtXalKj3zS9lRgBnCU7Ym2H6/rdz2qOlQ7Ud0o+GngDcDbgE+VNu8DltjeA9iD6uqrl9QHKGlSKZ44Y/nSRpVOIiJisAaTOO6yfWNZngLsW5a/D38vZjje9vVl/QXAawbQ71PAtLI8F7je9rKy3FPWvxE4plyq+xtgE2Db+o5Sqyoion0Gc+d4/fW7fa8fq2/YpGV+5trgFcCTALZXSOqLU8BJtq8a4rEiImKQBjPi2ErSXmX5SOCG2o2lCu6Dkl5dVr0b6Bt91NaqGoyrgA9KWhtA0naS1htCfxER0aTBjDhuB06UdB5wG/B14KS6Nu8BvlEeE3sn1XM4AM4v6x+nqlvVrHOpTlvdUsqR3AccsrIddtliHDNyQ15ERMs0ded4qVh7he2d2xVQq6VWVURE83LneEREtExTp6psLwJGzGgDYO7dS+g57coBtU2NqYiIVWvZiKP2Zr6VtDmlzHsMW18REdFaw32q6hSgVb/sW9lXREQMUNOJo7+yI3Vtvl7u3J4v6cyy7mSqOlXXSrq2rHujpF9LukXSJZLWb3C857Rp1FdERAyPwY44+is70ucTZTZ+V+C1kna1/RXgHuB15bGxE4DTgQNs705VjuT/1nbSX5v6vuqDS8mRiIj2Gewzx+vLjpxct/0dkiaV/jcDdgTm1LV5VVl/Y3VLBs8Dfj2INs9hezIwGaoHOQ3sLUVExEAMNnH0V3aEUnTwVGAP2w9KOh8Y06APAT+1/a6VHGcgbSIiYhgN9lTVysqObEhVt2qJpBcAb6rZVlty5CZgH0kvBZC0nqTt6o6zsjZDLV8SERGDMNjE0Vd2ZAGwEVXZEQBs3wrMAn4LfAe4sWa/ycA0Sdfavo/qwU7flTSH6hTU9rUHWUWbv/c1yPcQERGD0FTJERh5ZUdSciQionkpORIRES3T9OT4SCw7EhERrTPYq6pGjFXVqkp9qoiI5nT8VJWk8ZLqbyCs3f6r4YwnIiJWruOJAxjPc+88p+9xsbb3HvaIIiKiX91wquosYBtJs4FlwBPAg1SX3W4n6VHb60saBZwD7A/cVdqeZ3tqh+KOiFgjdUPiOA3Y2fZESfsBV5bXC+vaHUr12NgdgecDC4DzGnVYyp1MAhi94abtiToiYg3VDaeq6k1vkDQA9gUusb3C9l+Bfm/8sz3Zdq/t3tFjx7Ut0IiINVE3Jo7HOh1ARET0rxsSx0BrTt0IHCZpVKmBtV9bo4qIiIY6Psdh+35JN0qaBzwO3NtP00uB1wO3UU2O3wLkYRsREcOs6VpVnSRpfduPStoEmA7sU+Y7+pVaVRERzVtZraqOjziadIWk8VQPdPr3VSWNiIhovRGVOGzv1+w+qyo5Ui8lSCIiVq4bJscjImIEGZGJQ9KxkjbvdBwREWuiEZk4qJ4KmMQREdEBXTXHIelyYEtgDPBl4L/LTy9gqhIjd5XXF0t6HNjL9uOdiTgiYs3TVYkDOM72A5LWBW4GZgJb9D2mVtJ42w9J+hBwqu2G19mmVlVERPt026mqkyXdCtxENfJ4HrC1pK9KOhB4eCCdpFZVRET7dE3iKJVxD6A69bQbMAtYB9gNuA44Hji3U/FFRESlm05VjQMetL1U0vbAq4AJwCjbl0q6HZhS2g60vlVERLRYNyWOacDxkhYAt1OdrtoCuK48xAng4+XP84FvZHI8ImL4jahaVYORWlUREc1bWa2qrpnjiIiIkaGbTlW1xcpqVaUuVURE8zLiiIiIpgw6cUhaJGmCpJ7yEKYh9TPY/SMiYnhlxBEREU0ZUOKQdLmkmZLml3Ie/bXbWtIsSXvUre+R9FtJF0taIGmqpLE1TU6SdIukueUeDiRtXI47R9JNknYt618raXb5mSUp93NERAyjgY44jrP9CqrigieXR7c+i6SXUT0X/FjbNzfo42XA12zvQFU65ISabYtt7w58HTi1rDsTmGV7V+BfgAvL+lOBE21PBF5N9Zzy+lgmSZohacbypXkseUREKw00cdTXkNq2bvumwA+Bo2zf2k8fd9m+sSxPAfat2fY/5c+ZQE9Z3he4CMD2NcAmkjYEbgT+U9LJwHjbT9cfKLWqIiLaZ5WJo58aUmPqmi0B/sSzk0G9+jsNa18/Wf5cziouEbZ9FvB+YF3gxr5TWxERMTwGMuJoVEOq3lPA24BjJB3ZTz9bSdqrLB8J3LCK4/4SOAr+nrwW235Y0ja259r+PFXp9SSOiIhhNJAbABvVkHoO249JejPwU0mPAjOAc20fVJrcDpwo6TzgNqr5jJU5AzhP0hxgKfCesv4USa8DVgDzgZ+srJNdthjHjNzoFxHRMsNSq0pSD3BF3wOZhlNqVUVENC+1qiIiomWGpVaV7UXAsI82YOW1qiD1qiIimpURR0RENGXEJw5JozsdQ0TEmmTIiUPSMaUsyK2SLirlRa4p634uaavS7nxJ3yh3dN9RrsBC0rGSfijpOkm/k/TJmr6PljS9lBf5Zl+SkPSopP8oNyXu1TCwiIhoiyHNcUjaCTgd2Nv2YkkbAxcAF9i+QNJxwFeAQ8ouPcCewDbAtZJeWtbvSTUHshS4WdKVwGPAEcA+tpdJ+hrVfR0XAusBv7H9z0OJPyIimjfUyfH9gUtsLwaw/UC5ye/Qsv0i4As17X9gewXwO0l38szNez+1fT+ApP+hugP9aeAVVIkEqjvF/1baL6eqi9VQKcQ4CWD0hpsO8S1GRESt4X4CYH9lRxqtF9XI5eMN+nnC9vJ+D2JPBiYDrLPZtqv3Q9UjIobZUOc4rgHe3lctt5yq+hXwzrL9KKrSIX3eLmmUpG2AranuJgd4Qymjvi7Vaa0bgZ8Dh0t6fl/fkl48xHgjImKIhjTisD1f0meA6yUtpyqAeBLwbUkfAe4D3luzy5+A6cCGwPG2nyinoaZTnXp6ETDF9gwASacDV0saBSwDTgT+OJSYIyJiaIal5AhUV1VRlR2ZWrf+WKDX9ofacdyUHImIaF5KjkRERMsM2+S47WP7WX8+cP5wxREREUMz3FdVDbtV1arqT2pYRUQ0NqRTVZIWSZpQ7haf16qg6o5xrqQd29F3REQ0r+tHHLbf3+kYIiLiGQMecUi6XNJMSfPLndn11pJ0saQFkqZKGlv2WyRpQlnulXRdWT5D0gWSfinpj5IOlfQFSXMlTZO0dml3naTesvyopM+Uulg3SXrBUD+AiIhoTjOnqo6z/QqgFzi576a/Gi8DvmZ7B+Bh4IQB9LkNVdmStwBTgGtt7wI8DjSaZFgPuMn2bsAvgA806lTSpFJMccbypUsGEEZERAxUM4nj5FKN9iZgS2Dbuu132b6xLE+hqje1Kj+xvQyYC4ymer455XVPg/ZPAVeU5Zn9tMH2ZNu9tntHjx03gDAiImKgBjTHIWk/4ABgL9tLy+mmMXXN+qtD9TTPJKj6fZ4EsL1C0jI/czfiin5iq22zfKDxR0RE6wx0xDEOeLAkje2BVzVos1WpjAtwJHBDWV5EVeUW4LDBBhoREd1hoIljGtXk9wLgLKrTVfVuB04sbTYCvl7Wnwl8WdIMqlFCRESMYMNWq6pTUqsqIqJ5qVUVEREts9pPLjcqOZJyIhERg5cRR0RENKVrEoek8yUd3kT7ttXHioiI/nVN4oiIiJGhY4lD0jGS5pS6UxeV1a+R9CtJd/aNPlQ5W9K8UsfqiE7FHBERHZocl7QTcDqwt+3FkjYG/hPYjKpUyfbAj4CpwKHARGA3YAJws6RfrKL/ScAkgNEbbtqutxERsUbq1Ihjf+AS24sBbD9Q1l9ue4Xt24C+yrf7At+1vdz2vcD1wB4r6zy1qiIi2qfb5jierFlWx6KIiIh+dSpxXAO8va80ezlV1Z9fAkdIGi1pU+A1wPRhiDEiIhroyByH7fmSPgNcL2k5MGslzS8D9gJupaq4+1Hbf5XU0/ZAIyLiOVKrKiIiniO1qiIiomXWyFpVERGru3bW5MuIIyIimtJViaO/+lOSzpW0Y1leJGlCWf7VcMcYEbGmGxGnqmy/v5/1ew93LBERa7quGnEUa0m6WNICSVMljZV0naTnzO5LerQTAUZErMm6MXG8DPia7R2Ah4ETmu1A0iRJMyTNWL50ScsDjIhYk3Vj4rjL9o1leQpVraqmpFZVRET7dGPiqL8jcfW+QzEiYoTpxsSxlaS9yvKRwA2dDCYiIp6tGxPH7cCJkhYAGwFf73A8ERFRI7WqIiLiOVKrKiIiWmZE3AA4FKlVtXpqZx2eiFi5to84JB0r6Zx2HyciIoZHTlVFRERThpw4JB0tabqk2ZK+WR7x+l5Jd0iaDuxT03ZTSZdKurn87FPWnyHpAkm/lPRHSYdK+oKkuZKmSVq7tPu3st88SZMl5bnkERHDbEiJQ9IOwBHAPrYnAsuBo4EzqRLGvsCONbt8Gfii7T2Aw4Bza7ZtA+wPvIXqjvFrbe8CPA70ndA+x/YetncG1gXePJT4IyKieUOdHH898Arg5vLlf11gb+A62/cBSPo+sF1pfwCwY81AYUNJ65fln9heJmkuMBqYVtbPBXrK8uskfRQYC2wMzAf+tz4oSZOASQCjN9x0iG8xIiJqDTVxCLjA9sf/vkI6BDi0n/ajgFfZfuJZnVSJ5EkA2yskLfMzN5isoKqYOwb4GtBr+y5JZwBjGh3E9mRgMsA6m227et+oEhExzIY6x/Fz4HBJzweQtDEwC3itpE3K3MTba9pfDZzU90LSxCaO1ZckFpdRyuFDijwiIgZlSCMO27dJOh24WtIoYBlwInAG8GvgIWB2zS4nA/8laU459i+A4wd4rIckfQuYB/wVuHkosUdExOCk5EhERDxHSo5ERETLJHFERERTVvvE0VerKvWqIiJaYzhqVR0v6ZgG63skzWv38SMiorXaXh3X9jda0Y+ktWw/3Yq+IiJi8AY14pC0nqQrJd1a6kYdIWlRTX2p6ZJeWtqeIenUsvyKss+tVJft9vU3WtLZpQ7VHEn/VNbvV+pX/Qi4rdFxh/4RREREMwZ7qupA4B7bu5W6UX3lQZaU+lLnAF9qsN+3gZNs71a3/n1l3z2APYAPSHpJ2bY78H9sb7eS4z6LpEmSZkiasXzpkkG+xYiIaGSwiWMu8AZJn5f0att9v52/W/PnXrU7SBoPjLf9i7LqoprNbwSOkTQb+A2wCbBt2Tbd9sJVHPdZbE+23Wu7d/TYcYN8ixER0cigEoftO6hGAnOBT0v6t75Ntc2a6FJUI5GJ5ecltq8u2x4bwHEjImKYDHaOY3Ngqe0pwNlUv8yhKrHe9+eva/ex/RDwkKR9y6qjajZfBXyw5rkb20lar4njRkTEMBnsVVW7AGdLWkFVn+qDwFRgo1KH6kngXQ32ey9wniRTFTzscy5V6fRbysOZ7gMOGeBxIyJiGLWsVpWkRVQlzxe3pMMWSa2qiIjmpVZVRES0TMsSh+2ebhttwDMlRyIiojUy4oiIiKYMKXG0u95U6f/IdvUfERHN6/YRRw+QxBER0UVakThGS/qWpPmSrpa0rqRtJE2TNLPUmtoeQNL5kr5RyoHcIenNZX1PaXdL+dm79H0W8GpJsyV9WNIYSd8u9bBmSXpdC+KPiIgmtKI67rbAu2x/QNIPgMOo7tc43vbvJL0S+Bqwf2nfA+wJbANcW4oh/g14g+0nJG1LVbKkFzgNONV2X4L5Z8C2dynJ6GpJ29l+ojYgSZOASQCjN9y0BW8xIiL6tCJxLLQ9uyzPpEoMewOXVPfyAbBOTfsf2F4B/E7SncD2wELgHEkTgeXAdv0ca1/gqwC2fyvpj6XtnNpGticDkwHW2Wzb1fuh6hERw6wViePJmuXlwAuAh2xP7Kd9/S9yAx8G7gV2ozp99kT9ThER0R3aMTn+MLBQ0tsBVKkto/52SaMkbQNsDdwOjAP+UkYi7wZGl7aPABvU7PtLSo0rSdsBW5X9IyJimLTrqqqjgPeVBzbNB95as+1PwHTgJ1TzIE9QzYG8p7Tfnmcq4s4BlpcHN324tBslaS7wfeBY27UjnoiIaLOW1aoa0MGk84ErbE8drmOmVlVERPNSqyoiIlqmFZPjA2b72OE8Hgy9VtWisw5uYTQRESNfRhwREdGUjiYOSZtLmlqWJ0o6qJPxRETEqnU0cdi+x/bh5eVEIIkjIqLLDTlxSLq81KSaL2mSpNGlJtW8UlPqw6XdSyX9rFxae0upZ9VT2j0P+BRwRKlLdYSk9SSdJ2l6qUv11tLPTmXdbElzSomSiIgYJq2YHD/O9gOS1gVupio7soXtnQEkjS/tLgbOsn2ZpDFUSev5ALafkvRvVI+e/VDZ77PANbaPK31Ml/Qz4Hjgy7YvLglnNHVSqyoion1acarq5HLj3k3AlsDzgK0lfVXSgcDDkjagSiaXAdh+wvbSVfT7RuA0SbOB64AxVHeK/xr4F0kfA15s+/H6HW1Ptt1ru3f02HEteIsREdFnqA9y2g84ANjL9m7ALKqChrtR/bI/Hjh3sN0Dh9meWH62sr3A9neAtwCPAz+WtP/Ku4mIiFYa6ohjHPCg7aWlzPmrgAnAKNuXAqcDu9t+BPizpEMAJK0jaWxdX/V1qa4CTlIpsSvp5eXPrYE7bX8F+CGw6xDfQ0RENGGoiWMasJakBVQPXboJ2AK4rpximgJ8vLR9N9VprTnAr4AX1vV1LbBj3+Q48O/A2sAcSfPLa4B3APNK/zsDFw7xPURERBOGtVZVJ6RWVURE81KrKiIiWma1TxxDrVUVERHPttonjoiIaK0kjoiIaEoSR0RENKXjiaPUpLqy1LCaV+pUvULS9aUG1lWSNpO0lqSby02HSPqcpM90OPyIiDXOsD7IqR8HAvfYPhhA0jiq55G/1fZ95Z6Oz5SaVccCUyWdVPZ7ZaMOU6sqIqJ9uiFxzAX+Q9LngSuAB6lu7PtpuWl8NPAXANvzJV1U2u1l+6lGHdqeDEwGWGezbVfvG1UiIoZZxxOH7Tsk7U71LI5PA9cA823v1c8uuwAPUSrrRkTE8OqGOY7NgaW2pwBnU51+2lTSXmX72pJ2KsuHAhsDrwG+WlOyPSIihknHRxxUI4izJa0AlgEfBJ4GvlLmO9YCviTpXqp6WK+3fZekc4AvA+9ZaedbjGPGWQe39Q1ERKxJOp44bF9FVQm33msarNuuZr+vtC2oiIjoV8dPVUVExMjS8RFHu7WzVtWinAKLiDVQRhwREdGUJI6IiGhKxxOHpMtLaZH55Y5vJL1P0h2Spkv6VrmCCkmbSrq0lB65WdI+nY0+ImLN0w1zHMfZfkDSusDNkq4E/hXYneo55NcAt5a2Xwa+aPsGSVtRXY21Q32HKTkSEdE+3ZA4Tpb0trK8JdWzya+3/QCApEt45jLcA6ieS96374aS1rf9aG2HKTkSEdE+HU0cpdLtAVR1p5ZKug74LQ1GEcUo4FW2nxieCCMiol6n5zjGAQ+WpLE98CpgPeC1kjaStBZwWE37q4GT+l5Imjis0UZERMcTxzRgLUkLqMqJ3ATcDXwWmA7cCCwClpT2JwO9kuZIug04ftgjjohYw8nuvimAvnmLMuK4DDjP9mWD6au3t9czZsxobYAREas5STNt9zba1ukRR3/OkDQbmAcsBC7vcDwREVF0w1VVz2H71E7HEBERjXXriKNl2lmrKiJiTdSViUPSeEknNLstIiLarysTBzAeeE5yKJPlDbdFRMTw6Mo5DqpLc7cpE+TLgCeAB4HtgVtqtv3U9kc6F2ZExJqnWxPHacDOtieWu8uvLK8XSurp29bfzqlVFRHRPt16qqredNsLB9rY9mTbvbZ7R48d1864Ig1IAc8AAAUhSURBVCLWOCMlcTzW6QAiIqLSrYnjEWCDQWyLiIg268o5Dtv3S7pR0jzgceDefrb9JJPjERHDqytrVbVSalVFRDRvJNaqioiILpXEERERTUniiIiIpiRxREREU5I4IiKiKUkcERHRlCSOiIhoShJHREQ0JYkjIiKastrfOS7pEeD2TsfRjwnA4k4H0Y9uja1b44LENljdGlu3xgXDE9uLbTd8LkVX1qpqsdv7u22+0yTNSGzN6da4ILENVrfG1q1xQedjy6mqiIhoShJHREQ0ZU1IHJM7HcBKJLbmdWtckNgGq1tj69a4oMOxrfaT4xER0VprwogjIiJaKIkjIiKaMqITh6QDJd0u6feSTmuwfR1J3y/bfyOpp2bbx8v62yX9QzfEJWkTSddKelTSOa2MqQWxvUHSTElzy5/7d1Fse0qaXX5ulfS2bomtZvtW5e/11G6IS1KPpMdrPrdvtDKuocRWtu0q6deS5pd/c2O6ITZJR9V8ZrMlrZA0sUtiW1vSBeXzWiDp462M61lsj8gfYDTwB2Br4HnArcCOdW1OAL5Rlt8JfL8s71jarwO8pPQzugviWg/YFzgeOKfLPrOXA5uX5Z2Bu7sotrHAWmV5M+Bvfa87HVvN9qnAJcCp3RAX0APMa/W/sRbFthYwB9itvN6kVf8/W/X3WdbvAvyhiz63I4Hv1fyfWAT0tOPvdySPOPYEfm/7TttPAd8D3lrX5q3ABWV5KvB6SSrrv2f7SdsLgd+X/joal+3HbN8APNGiWFoZ2yzb95T184F1Ja3TJbEttf10WT8GaPUVH0P5t4akQ4CFVJ9b18TVZkOJ7Y3AHNu3Ati+3/byLomt1rvKvq00lNgMrCdpLWBd4Cng4RbHB4zsU1VbAHfVvP5zWdewTfnFsoTq28tA9u1EXO3WqtgOA26x/WS3xCbplZLmA3OB42sSSUdjk7Q+8DHgzBbGM+S4yraXSJol6XpJr+6i2LYDLOkqSbdI+mgXxVbrCOC7XRTbVOAx4C/An4D/Z/uBFscHrBklR6KFJO0EfJ7qW2HXsP0bYCdJOwAXSPqJ7XaN3JpxBvBF248Ozxf9AfsLsJXt+yW9Arhc0k622/INtUlrUZ2y3QNYCvxc0kzbP+9sWM+Q9Epgqe15nY6lxp7AcmBzYCPgl5J+ZvvOVh9oJI847ga2rHn9orKuYZsyfBsH3D/AfTsRV7sNKTZJLwIuA46x/Yduiq2P7QXAo1TzMN0Q2yuBL0haBJwC/IukD3U6rnKa9n4A2zOpzqtv16K4hhQb1bfsX9hebHsp8GNg9y6Jrc87af1oY6ixHQlMs73M9t+AG4H21LNqx8TJcPxQfSu5k2pyu28Saae6Nify7EmkH5TlnXj25PidtG5yfNBx1Ww/lvZMjg/lMxtf2h/ahX+fL+GZyfEXA/cAE7ohtro2Z9DayfGhfGab9v2bp5qIvRvYuEti2wi4hXLRA/Az4OBuiK28HlU+r6277P/Bx4Bvl+X1gNuAXVsdo+2RmzjKh3MQcAfVt6VPlHWfAt5SlsdQXcnye2B67V808Imy3+3Am7oorkXAA1Tfmv9M3RUVnYoNOJ3q/Onsmp/nd0ls76aaeJ5dfuEc0k3/1mr6OIMWJo4hfmaH1X1m/9hNnxlwdIlvHvCFLottP+CmVsfUgr/T9cv6+VRJ4yPtijElRyIioikjeY4jIiI6IIkjIiKaksQRERFNSeKIiIimJHFERERTkjgiIqIpSRwREdGU/w8SRrg2nOTlqgAAAABJRU5ErkJggg==\n",
      "text/plain": [
       "<Figure size 432x288 with 1 Axes>"
      ]
     },
     "metadata": {
      "needs_background": "light"
     },
     "output_type": "display_data"
    }
   ],
   "source": [
    "vimps = np.array(forest.vimp(model).rx('importance')[0])\n",
    "\n",
    "y = np.arange(len(vimps))\n",
    "plt.barh(y, np.abs(vimps))\n",
    "plt.yticks(y, df_train.drop(['time', 'status'], axis=1).columns)\n",
    "plt.title(\"VIMP (absolute value)\")\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "2YGhK2xwjkiA"
   },
   "source": [
    "### Question:\n",
    "\n",
    "How does the variable importance compare to that of the Cox model? Which variable is important in both models? Which variable is important in the random survival forest but not in the Cox model? You should see that `edema` is important in both the random survival forest and the Cox model. You should also see that `bili` is important in the random survival forest but not the Cox model ."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Congratulations!\n",
    "\n",
    "You've finished the last assignment in course 2! Take a minute to look back at the analysis you've done over the last four assignments. You've done a great job!"
   ]
  }
 ],
 "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.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}