{ "cells": [ { "cell_type": "code", "execution_count": 1, "id": "08bd379a", "metadata": {}, "outputs": [], "source": [ "import torch\n", "from models import TimeAwareGPT2\n", "from utils import load_model\n", "from tqdm import tqdm\n", "import pandas as pd\n", "import numpy as np\n", "import textwrap\n", "import matplotlib.pyplot as plt\n", "\n", "plt.rcParams['figure.facecolor'] = 'white'\n", "plt.rcParams.update({'axes.grid': True,\n", " 'grid.linestyle': ':',\n", " 'axes.spines.bottom': False,\n", " 'axes.spines.left': False,\n", " 'axes.spines.right': False,\n", " 'axes.spines.top': False})\n", "plt.rcParams['figure.dpi'] = 72\n", "plt.rcParams['pdf.fonttype'] = 42\n", "\n", "#Green\n", "light_male = '#BAEBE3'\n", "normal_male = '#0FB8A1'\n", "dark_male = '#00574A'\n", "\n", "\n", "#Purple\n", "light_female = '#DEC7FF'\n", "normal_female = '#8520F1'\n", "dark_female = '#7A00BF'\n", "\n", " \n", "delphi_labels = pd.read_csv('delphi_labels_chapters_colours_icd.csv')" ] }, { "cell_type": "code", "execution_count": 2, "id": "1d8375ab", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", " | name | \n", "ICD-10 Chapter (short) | \n", "
---|---|---|
46 | \n", "A41 Other septicaemia | \n", "I. Infectious Diseases | \n", "
95 | \n", "B01 Varicella [chickenpox] | \n", "I. Infectious Diseases | \n", "
1168 | \n", "C25 Malignant neoplasm of pancreas | \n", "II. Neoplasms | \n", "
1188 | \n", "C50 Malignant neoplasm of breast | \n", "II. Neoplasms | \n", "
374 | \n", "G30 Alzheimer's disease | \n", "VI. Nervous System Diseases | \n", "
214 | \n", "E10 Insulin-dependent diabetes mellitus | \n", "IV. Metabolic Diseases | \n", "
305 | \n", "F32 Depressive episode | \n", "V. Mental Disorders | \n", "
505 | \n", "I21 Acute myocardial infarction | \n", "IX. Circulatory Diseases | \n", "
603 | \n", "J45 Asthma | \n", "X. Respiratory Diseases | \n", "
1269 | \n", "Death | \n", "Death | \n", "