embeddinggenerationandprediction.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Embedding Generation and Prediction"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Generate Embeddings for All Possible Combinations"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "#set variables according to docker instance\n",
    "uri = \"bolt://localhost:7687\"\n",
    "username = \"neo4j\"\n",
    "password = \"password\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "#connect with driver\n",
    "from neo4j import GraphDatabase\n",
    "with GraphDatabase.driver(uri,auth=(username,password)) as driver:\n",
    "  driver.verify_connectivity()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "#setup GDS\n",
    "from re import U\n",
    "from graphdatascience import GraphDataScience\n",
    "\n",
    "gds = GraphDataScience(uri, auth=(username,password))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "#method that clears all graphs in memory - can only have a specific amount according to memory\n",
    "def clearGraphs():\n",
    "    driver.execute_query(\n",
    "        \"\"\"\n",
    "        CALL gds.graph.list()\n",
    "        YIELD graphName\n",
    "        WITH collect(graphName) AS graphs\n",
    "        UNWIND graphs AS graph\n",
    "        CALL gds.graph.drop(graph) YIELD graphName\n",
    "        RETURN graphName\n",
    "        \"\"\"\n",
    "    )"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [],
   "source": [
    "import itertools\n",
    "\n",
    "# Define the variations for each field\n",
    "fields_variations = {\n",
    "    \"pregnancies\" : [\"low pregnancies\", \"normal pregnancies\", \"high pregnancies\"],\n",
    "    \"glucose\" : [\"low glucose\", \"normal glucose\", \"high glucose\"],\n",
    "    \"blood_pressure\" : [\"low blood pressure\", \"normal blood pressure\", \"high blood pressure\"],\n",
    "    \"skin_thickness\" : [\"low skin thickness\", \"normal skin thickness\", \"high skin thickness\"],\n",
    "    \"insulin\" : [\"low insulin\", \"normal insulin\", \"high insulin\"],\n",
    "    \"bmi\" : [\"low bmi\", \"normal bmi\", \"high bmi\"],\n",
    "    \"diabetes_pedigree_function\" : [\"low diabetes pedigree function\", \"normal diabetes pedigree function\", \"high diabetes pedigree function\"],\n",
    "    \"age\" : [\"low age\", \"normal age\", \"high age\"]\n",
    "}\n",
    "\n",
    "relationships = {\n",
    "    \"low pregnancies\": \"HAS_PREGNANCIES\",\n",
    "    \"normal pregnancies\": \"HAS_PREGNANCIES\",\n",
    "    \"high pregnancies\": \"HAS_PREGNANCIES\",\n",
    "    \"low glucose\": \"HAS_GLUCOSE\",\n",
    "    \"normal glucose\": \"HAS_GLUCOSE\",\n",
    "    \"high glucose\": \"HAS_GLUCOSE\",\n",
    "    \"low blood pressure\": \"HAS_BLOOD_PRESSURE\",\n",
    "    \"normal blood pressure\": \"HAS_BLOOD_PRESSURE\",\n",
    "    \"high blood pressure\": \"HAS_BLOOD_PRESSURE\",\n",
    "    \"low skin thickness\": \"HAS_SKIN_THICKNESS\",\n",
    "    \"normal skin thickness\": \"HAS_SKIN_THICKNESS\",\n",
    "    \"high skin thickness\": \"HAS_SKIN_THICKNESS\",\n",
    "    \"low insulin\": \"HAS_INSULIN\",\n",
    "    \"normal insulin\": \"HAS_INSULIN\",\n",
    "    \"high insulin\": \"HAS_INSULIN\",\n",
    "    \"low bmi\": \"HAS_BMI\",\n",
    "    \"normal bmi\": \"HAS_BMI\",\n",
    "    \"high bmi\": \"HAS_BMI\",\n",
    "    \"low diabetes pedigree function\": \"HAS_DIABETES_PEDIGREE_FUNCTION\",\n",
    "    \"normal diabetes pedigree function\": \"HAS_DIABETES_PEDIGREE_FUNCTION\",\n",
    "    \"high diabetes pedigree function\": \"HAS_DIABETES_PEDIGREE_FUNCTION\",\n",
    "    \"low age\": \"HAS_AGE\",\n",
    "    \"normal age\": \"HAS_AGE\",\n",
    "    \"high age\": \"HAS_AGE\"\n",
    "}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "metadata": {},
   "outputs": [],
   "source": [
    "def generate_combinations(fields_variations, combination_size):\n",
    "    # Get all field names\n",
    "    field_names = list(fields_variations.keys())\n",
    "\n",
    "    # Get all possible combinations of the specified size\n",
    "    field_combinations = list(itertools.combinations(field_names, combination_size))\n",
    "\n",
    "    # Function to generate all combinations of variations for a given set of fields\n",
    "    def get_variations(fields):\n",
    "        variations = [fields_variations[field] for field in fields]\n",
    "        return list(itertools.product(*variations))\n",
    "\n",
    "    # Generate all combinations for each set of fields\n",
    "    all_combinations = []\n",
    "    for fields in field_combinations:\n",
    "        combinations = get_variations(fields)\n",
    "        all_combinations.extend(combinations)\n",
    "\n",
    "    # Convert to list of lists\n",
    "    combinations_list = [list(combination) for combination in all_combinations]\n",
    "\n",
    "    return combinations_list"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[['low pregnancies', 'low glucose', 'low blood pressure'],\n",
       " ['low pregnancies', 'low glucose', 'normal blood pressure'],\n",
       " ['low pregnancies', 'low glucose', 'high blood pressure'],\n",
       " ['low pregnancies', 'normal glucose', 'low blood pressure'],\n",
       " ['low pregnancies', 'normal glucose', 'normal blood pressure'],\n",
       " ['low pregnancies', 'normal glucose', 'high blood pressure'],\n",
       " ['low pregnancies', 'high glucose', 'low blood pressure'],\n",
       " ['low pregnancies', 'high glucose', 'normal blood pressure'],\n",
       " ['low pregnancies', 'high glucose', 'high blood pressure'],\n",
       " ['normal pregnancies', 'low glucose', 'low blood pressure'],\n",
       " ['normal pregnancies', 'low glucose', 'normal blood pressure'],\n",
       " ['normal pregnancies', 'low glucose', 'high blood pressure'],\n",
       " ['normal pregnancies', 'normal glucose', 'low blood pressure'],\n",
       " ['normal pregnancies', 'normal glucose', 'normal blood pressure'],\n",
       " ['normal pregnancies', 'normal glucose', 'high blood pressure'],\n",
       " ['normal pregnancies', 'high glucose', 'low blood pressure'],\n",
       " ['normal pregnancies', 'high glucose', 'normal blood pressure'],\n",
       " ['normal pregnancies', 'high glucose', 'high blood pressure'],\n",
       " ['high pregnancies', 'low glucose', 'low blood pressure'],\n",
       " ['high pregnancies', 'low glucose', 'normal blood pressure'],\n",
       " ['high pregnancies', 'low glucose', 'high blood pressure'],\n",
       " ['high pregnancies', 'normal glucose', 'low blood pressure'],\n",
       " ['high pregnancies', 'normal glucose', 'normal blood pressure'],\n",
       " ['high pregnancies', 'normal glucose', 'high blood pressure'],\n",
       " ['high pregnancies', 'high glucose', 'low blood pressure'],\n",
       " ['high pregnancies', 'high glucose', 'normal blood pressure'],\n",
       " ['high pregnancies', 'high glucose', 'high blood pressure'],\n",
       " ['low pregnancies', 'low glucose', 'low skin thickness'],\n",
       " ['low pregnancies', 'low glucose', 'normal skin thickness'],\n",
       " ['low pregnancies', 'low glucose', 'high skin thickness'],\n",
       " ['low pregnancies', 'normal glucose', 'low skin thickness'],\n",
       " ['low pregnancies', 'normal glucose', 'normal skin thickness'],\n",
       " ['low pregnancies', 'normal glucose', 'high skin thickness'],\n",
       " ['low pregnancies', 'high glucose', 'low skin thickness'],\n",
       " ['low pregnancies', 'high glucose', 'normal skin thickness'],\n",
       " ['low pregnancies', 'high glucose', 'high skin thickness'],\n",
       " ['normal pregnancies', 'low glucose', 'low skin thickness'],\n",
       " ['normal pregnancies', 'low glucose', 'normal skin thickness'],\n",
       " ['normal pregnancies', 'low glucose', 'high skin thickness'],\n",
       " ['normal pregnancies', 'normal glucose', 'low skin thickness'],\n",
       " ['normal pregnancies', 'normal glucose', 'normal skin thickness'],\n",
       " ['normal pregnancies', 'normal glucose', 'high skin thickness'],\n",
       " ['normal pregnancies', 'high glucose', 'low skin thickness'],\n",
       " ['normal pregnancies', 'high glucose', 'normal skin thickness'],\n",
       " ['normal pregnancies', 'high glucose', 'high skin thickness'],\n",
       " ['high pregnancies', 'low glucose', 'low skin thickness'],\n",
       " ['high pregnancies', 'low glucose', 'normal skin thickness'],\n",
       " ['high pregnancies', 'low glucose', 'high skin thickness'],\n",
       " ['high pregnancies', 'normal glucose', 'low skin thickness'],\n",
       " ['high pregnancies', 'normal glucose', 'normal skin thickness'],\n",
       " ['high pregnancies', 'normal glucose', 'high skin thickness'],\n",
       " ['high pregnancies', 'high glucose', 'low skin thickness'],\n",
       " ['high pregnancies', 'high glucose', 'normal skin thickness'],\n",
       " ['high pregnancies', 'high glucose', 'high skin thickness'],\n",
       " ['low pregnancies', 'low glucose', 'low insulin'],\n",
       " ['low pregnancies', 'low glucose', 'normal insulin'],\n",
       " ['low pregnancies', 'low glucose', 'high insulin'],\n",
       " ['low pregnancies', 'normal glucose', 'low insulin'],\n",
       " ['low pregnancies', 'normal glucose', 'normal insulin'],\n",
       " ['low pregnancies', 'normal glucose', 'high insulin'],\n",
       " ['low pregnancies', 'high glucose', 'low insulin'],\n",
       " ['low pregnancies', 'high glucose', 'normal insulin'],\n",
       " ['low pregnancies', 'high glucose', 'high insulin'],\n",
       " ['normal pregnancies', 'low glucose', 'low insulin'],\n",
       " ['normal pregnancies', 'low glucose', 'normal insulin'],\n",
       " ['normal pregnancies', 'low glucose', 'high insulin'],\n",
       " ['normal pregnancies', 'normal glucose', 'low insulin'],\n",
       " ['normal pregnancies', 'normal glucose', 'normal insulin'],\n",
       " ['normal pregnancies', 'normal glucose', 'high insulin'],\n",
       " ['normal pregnancies', 'high glucose', 'low insulin'],\n",
       " ['normal pregnancies', 'high glucose', 'normal insulin'],\n",
       " ['normal pregnancies', 'high glucose', 'high insulin'],\n",
       " ['high pregnancies', 'low glucose', 'low insulin'],\n",
       " ['high pregnancies', 'low glucose', 'normal insulin'],\n",
       " ['high pregnancies', 'low glucose', 'high insulin'],\n",
       " ['high pregnancies', 'normal glucose', 'low insulin'],\n",
       " ['high pregnancies', 'normal glucose', 'normal insulin'],\n",
       " ['high pregnancies', 'normal glucose', 'high insulin'],\n",
       " ['high pregnancies', 'high glucose', 'low insulin'],\n",
       " ['high pregnancies', 'high glucose', 'normal insulin'],\n",
       " ['high pregnancies', 'high glucose', 'high insulin'],\n",
       " ['low pregnancies', 'low glucose', 'low bmi'],\n",
       " ['low pregnancies', 'low glucose', 'normal bmi'],\n",
       " ['low pregnancies', 'low glucose', 'high bmi'],\n",
       " ['low pregnancies', 'normal glucose', 'low bmi'],\n",
       " ['low pregnancies', 'normal glucose', 'normal bmi'],\n",
       " ['low pregnancies', 'normal glucose', 'high bmi'],\n",
       " ['low pregnancies', 'high glucose', 'low bmi'],\n",
       " ['low pregnancies', 'high glucose', 'normal bmi'],\n",
       " ['low pregnancies', 'high glucose', 'high bmi'],\n",
       " ['normal pregnancies', 'low glucose', 'low bmi'],\n",
       " ['normal pregnancies', 'low glucose', 'normal bmi'],\n",
       " ['normal pregnancies', 'low glucose', 'high bmi'],\n",
       " ['normal pregnancies', 'normal glucose', 'low bmi'],\n",
       " ['normal pregnancies', 'normal glucose', 'normal bmi'],\n",
       " ['normal pregnancies', 'normal glucose', 'high bmi'],\n",
       " ['normal pregnancies', 'high glucose', 'low bmi'],\n",
       " ['normal pregnancies', 'high glucose', 'normal bmi'],\n",
       " ['normal pregnancies', 'high glucose', 'high bmi'],\n",
       " ['high pregnancies', 'low glucose', 'low bmi'],\n",
       " ['high pregnancies', 'low glucose', 'normal bmi'],\n",
       " ['high pregnancies', 'low glucose', 'high bmi'],\n",
       " ['high pregnancies', 'normal glucose', 'low bmi'],\n",
       " ['high pregnancies', 'normal glucose', 'normal bmi'],\n",
       " ['high pregnancies', 'normal glucose', 'high bmi'],\n",
       " ['high pregnancies', 'high glucose', 'low bmi'],\n",
       " ['high pregnancies', 'high glucose', 'normal bmi'],\n",
       " ['high pregnancies', 'high glucose', 'high bmi'],\n",
       " ['low pregnancies', 'low glucose', 'low diabetes pedigree function'],\n",
       " ['low pregnancies', 'low glucose', 'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'low glucose', 'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'normal glucose', 'low diabetes pedigree function'],\n",
       " ['low pregnancies', 'normal glucose', 'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'normal glucose', 'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'high glucose', 'low diabetes pedigree function'],\n",
       " ['low pregnancies', 'high glucose', 'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'high glucose', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies', 'low glucose', 'low diabetes pedigree function'],\n",
       " ['normal pregnancies', 'low glucose', 'normal diabetes pedigree function'],\n",
       " ['normal pregnancies', 'low glucose', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies', 'normal glucose', 'low diabetes pedigree function'],\n",
       " ['normal pregnancies', 'normal glucose', 'normal diabetes pedigree function'],\n",
       " ['normal pregnancies', 'normal glucose', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies', 'high glucose', 'low diabetes pedigree function'],\n",
       " ['normal pregnancies', 'high glucose', 'normal diabetes pedigree function'],\n",
       " ['normal pregnancies', 'high glucose', 'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'low glucose', 'low diabetes pedigree function'],\n",
       " ['high pregnancies', 'low glucose', 'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'low glucose', 'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'normal glucose', 'low diabetes pedigree function'],\n",
       " ['high pregnancies', 'normal glucose', 'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'normal glucose', 'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'high glucose', 'low diabetes pedigree function'],\n",
       " ['high pregnancies', 'high glucose', 'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'high glucose', 'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'low glucose', 'low age'],\n",
       " ['low pregnancies', 'low glucose', 'normal age'],\n",
       " ['low pregnancies', 'low glucose', 'high age'],\n",
       " ['low pregnancies', 'normal glucose', 'low age'],\n",
       " ['low pregnancies', 'normal glucose', 'normal age'],\n",
       " ['low pregnancies', 'normal glucose', 'high age'],\n",
       " ['low pregnancies', 'high glucose', 'low age'],\n",
       " ['low pregnancies', 'high glucose', 'normal age'],\n",
       " ['low pregnancies', 'high glucose', 'high age'],\n",
       " ['normal pregnancies', 'low glucose', 'low age'],\n",
       " ['normal pregnancies', 'low glucose', 'normal age'],\n",
       " ['normal pregnancies', 'low glucose', 'high age'],\n",
       " ['normal pregnancies', 'normal glucose', 'low age'],\n",
       " ['normal pregnancies', 'normal glucose', 'normal age'],\n",
       " ['normal pregnancies', 'normal glucose', 'high age'],\n",
       " ['normal pregnancies', 'high glucose', 'low age'],\n",
       " ['normal pregnancies', 'high glucose', 'normal age'],\n",
       " ['normal pregnancies', 'high glucose', 'high age'],\n",
       " ['high pregnancies', 'low glucose', 'low age'],\n",
       " ['high pregnancies', 'low glucose', 'normal age'],\n",
       " ['high pregnancies', 'low glucose', 'high age'],\n",
       " ['high pregnancies', 'normal glucose', 'low age'],\n",
       " ['high pregnancies', 'normal glucose', 'normal age'],\n",
       " ['high pregnancies', 'normal glucose', 'high age'],\n",
       " ['high pregnancies', 'high glucose', 'low age'],\n",
       " ['high pregnancies', 'high glucose', 'normal age'],\n",
       " ['high pregnancies', 'high glucose', 'high age'],\n",
       " ['low pregnancies', 'low blood pressure', 'low skin thickness'],\n",
       " ['low pregnancies', 'low blood pressure', 'normal skin thickness'],\n",
       " ['low pregnancies', 'low blood pressure', 'high skin thickness'],\n",
       " ['low pregnancies', 'normal blood pressure', 'low skin thickness'],\n",
       " ['low pregnancies', 'normal blood pressure', 'normal skin thickness'],\n",
       " ['low pregnancies', 'normal blood pressure', 'high skin thickness'],\n",
       " ['low pregnancies', 'high blood pressure', 'low skin thickness'],\n",
       " ['low pregnancies', 'high blood pressure', 'normal skin thickness'],\n",
       " ['low pregnancies', 'high blood pressure', 'high skin thickness'],\n",
       " ['normal pregnancies', 'low blood pressure', 'low skin thickness'],\n",
       " ['normal pregnancies', 'low blood pressure', 'normal skin thickness'],\n",
       " ['normal pregnancies', 'low blood pressure', 'high skin thickness'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'low skin thickness'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'normal skin thickness'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'high skin thickness'],\n",
       " ['normal pregnancies', 'high blood pressure', 'low skin thickness'],\n",
       " ['normal pregnancies', 'high blood pressure', 'normal skin thickness'],\n",
       " ['normal pregnancies', 'high blood pressure', 'high skin thickness'],\n",
       " ['high pregnancies', 'low blood pressure', 'low skin thickness'],\n",
       " ['high pregnancies', 'low blood pressure', 'normal skin thickness'],\n",
       " ['high pregnancies', 'low blood pressure', 'high skin thickness'],\n",
       " ['high pregnancies', 'normal blood pressure', 'low skin thickness'],\n",
       " ['high pregnancies', 'normal blood pressure', 'normal skin thickness'],\n",
       " ['high pregnancies', 'normal blood pressure', 'high skin thickness'],\n",
       " ['high pregnancies', 'high blood pressure', 'low skin thickness'],\n",
       " ['high pregnancies', 'high blood pressure', 'normal skin thickness'],\n",
       " ['high pregnancies', 'high blood pressure', 'high skin thickness'],\n",
       " ['low pregnancies', 'low blood pressure', 'low insulin'],\n",
       " ['low pregnancies', 'low blood pressure', 'normal insulin'],\n",
       " ['low pregnancies', 'low blood pressure', 'high insulin'],\n",
       " ['low pregnancies', 'normal blood pressure', 'low insulin'],\n",
       " ['low pregnancies', 'normal blood pressure', 'normal insulin'],\n",
       " ['low pregnancies', 'normal blood pressure', 'high insulin'],\n",
       " ['low pregnancies', 'high blood pressure', 'low insulin'],\n",
       " ['low pregnancies', 'high blood pressure', 'normal insulin'],\n",
       " ['low pregnancies', 'high blood pressure', 'high insulin'],\n",
       " ['normal pregnancies', 'low blood pressure', 'low insulin'],\n",
       " ['normal pregnancies', 'low blood pressure', 'normal insulin'],\n",
       " ['normal pregnancies', 'low blood pressure', 'high insulin'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'low insulin'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'normal insulin'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'high insulin'],\n",
       " ['normal pregnancies', 'high blood pressure', 'low insulin'],\n",
       " ['normal pregnancies', 'high blood pressure', 'normal insulin'],\n",
       " ['normal pregnancies', 'high blood pressure', 'high insulin'],\n",
       " ['high pregnancies', 'low blood pressure', 'low insulin'],\n",
       " ['high pregnancies', 'low blood pressure', 'normal insulin'],\n",
       " ['high pregnancies', 'low blood pressure', 'high insulin'],\n",
       " ['high pregnancies', 'normal blood pressure', 'low insulin'],\n",
       " ['high pregnancies', 'normal blood pressure', 'normal insulin'],\n",
       " ['high pregnancies', 'normal blood pressure', 'high insulin'],\n",
       " ['high pregnancies', 'high blood pressure', 'low insulin'],\n",
       " ['high pregnancies', 'high blood pressure', 'normal insulin'],\n",
       " ['high pregnancies', 'high blood pressure', 'high insulin'],\n",
       " ['low pregnancies', 'low blood pressure', 'low bmi'],\n",
       " ['low pregnancies', 'low blood pressure', 'normal bmi'],\n",
       " ['low pregnancies', 'low blood pressure', 'high bmi'],\n",
       " ['low pregnancies', 'normal blood pressure', 'low bmi'],\n",
       " ['low pregnancies', 'normal blood pressure', 'normal bmi'],\n",
       " ['low pregnancies', 'normal blood pressure', 'high bmi'],\n",
       " ['low pregnancies', 'high blood pressure', 'low bmi'],\n",
       " ['low pregnancies', 'high blood pressure', 'normal bmi'],\n",
       " ['low pregnancies', 'high blood pressure', 'high bmi'],\n",
       " ['normal pregnancies', 'low blood pressure', 'low bmi'],\n",
       " ['normal pregnancies', 'low blood pressure', 'normal bmi'],\n",
       " ['normal pregnancies', 'low blood pressure', 'high bmi'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'low bmi'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'normal bmi'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'high bmi'],\n",
       " ['normal pregnancies', 'high blood pressure', 'low bmi'],\n",
       " ['normal pregnancies', 'high blood pressure', 'normal bmi'],\n",
       " ['normal pregnancies', 'high blood pressure', 'high bmi'],\n",
       " ['high pregnancies', 'low blood pressure', 'low bmi'],\n",
       " ['high pregnancies', 'low blood pressure', 'normal bmi'],\n",
       " ['high pregnancies', 'low blood pressure', 'high bmi'],\n",
       " ['high pregnancies', 'normal blood pressure', 'low bmi'],\n",
       " ['high pregnancies', 'normal blood pressure', 'normal bmi'],\n",
       " ['high pregnancies', 'normal blood pressure', 'high bmi'],\n",
       " ['high pregnancies', 'high blood pressure', 'low bmi'],\n",
       " ['high pregnancies', 'high blood pressure', 'normal bmi'],\n",
       " ['high pregnancies', 'high blood pressure', 'high bmi'],\n",
       " ['low pregnancies', 'low blood pressure', 'low diabetes pedigree function'],\n",
       " ['low pregnancies',\n",
       "  'low blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'low blood pressure', 'high diabetes pedigree function'],\n",
       " ['low pregnancies',\n",
       "  'normal blood pressure',\n",
       "  'low diabetes pedigree function'],\n",
       " ['low pregnancies',\n",
       "  'normal blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['low pregnancies',\n",
       "  'normal blood pressure',\n",
       "  'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'high blood pressure', 'low diabetes pedigree function'],\n",
       " ['low pregnancies',\n",
       "  'high blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'high blood pressure', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'low blood pressure',\n",
       "  'low diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'low blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'low blood pressure',\n",
       "  'high diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'normal blood pressure',\n",
       "  'low diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'normal blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'normal blood pressure',\n",
       "  'high diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'high blood pressure',\n",
       "  'low diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'high blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'high blood pressure',\n",
       "  'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'low blood pressure', 'low diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'low blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'low blood pressure', 'high diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'normal blood pressure',\n",
       "  'low diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'normal blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'normal blood pressure',\n",
       "  'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'high blood pressure', 'low diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'high blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'high blood pressure',\n",
       "  'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'low blood pressure', 'low age'],\n",
       " ['low pregnancies', 'low blood pressure', 'normal age'],\n",
       " ['low pregnancies', 'low blood pressure', 'high age'],\n",
       " ['low pregnancies', 'normal blood pressure', 'low age'],\n",
       " ['low pregnancies', 'normal blood pressure', 'normal age'],\n",
       " ['low pregnancies', 'normal blood pressure', 'high age'],\n",
       " ['low pregnancies', 'high blood pressure', 'low age'],\n",
       " ['low pregnancies', 'high blood pressure', 'normal age'],\n",
       " ['low pregnancies', 'high blood pressure', 'high age'],\n",
       " ['normal pregnancies', 'low blood pressure', 'low age'],\n",
       " ['normal pregnancies', 'low blood pressure', 'normal age'],\n",
       " ['normal pregnancies', 'low blood pressure', 'high age'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'low age'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'normal age'],\n",
       " ['normal pregnancies', 'normal blood pressure', 'high age'],\n",
       " ['normal pregnancies', 'high blood pressure', 'low age'],\n",
       " ['normal pregnancies', 'high blood pressure', 'normal age'],\n",
       " ['normal pregnancies', 'high blood pressure', 'high age'],\n",
       " ['high pregnancies', 'low blood pressure', 'low age'],\n",
       " ['high pregnancies', 'low blood pressure', 'normal age'],\n",
       " ['high pregnancies', 'low blood pressure', 'high age'],\n",
       " ['high pregnancies', 'normal blood pressure', 'low age'],\n",
       " ['high pregnancies', 'normal blood pressure', 'normal age'],\n",
       " ['high pregnancies', 'normal blood pressure', 'high age'],\n",
       " ['high pregnancies', 'high blood pressure', 'low age'],\n",
       " ['high pregnancies', 'high blood pressure', 'normal age'],\n",
       " ['high pregnancies', 'high blood pressure', 'high age'],\n",
       " ['low pregnancies', 'low skin thickness', 'low insulin'],\n",
       " ['low pregnancies', 'low skin thickness', 'normal insulin'],\n",
       " ['low pregnancies', 'low skin thickness', 'high insulin'],\n",
       " ['low pregnancies', 'normal skin thickness', 'low insulin'],\n",
       " ['low pregnancies', 'normal skin thickness', 'normal insulin'],\n",
       " ['low pregnancies', 'normal skin thickness', 'high insulin'],\n",
       " ['low pregnancies', 'high skin thickness', 'low insulin'],\n",
       " ['low pregnancies', 'high skin thickness', 'normal insulin'],\n",
       " ['low pregnancies', 'high skin thickness', 'high insulin'],\n",
       " ['normal pregnancies', 'low skin thickness', 'low insulin'],\n",
       " ['normal pregnancies', 'low skin thickness', 'normal insulin'],\n",
       " ['normal pregnancies', 'low skin thickness', 'high insulin'],\n",
       " ['normal pregnancies', 'normal skin thickness', 'low insulin'],\n",
       " ['normal pregnancies', 'normal skin thickness', 'normal insulin'],\n",
       " ['normal pregnancies', 'normal skin thickness', 'high insulin'],\n",
       " ['normal pregnancies', 'high skin thickness', 'low insulin'],\n",
       " ['normal pregnancies', 'high skin thickness', 'normal insulin'],\n",
       " ['normal pregnancies', 'high skin thickness', 'high insulin'],\n",
       " ['high pregnancies', 'low skin thickness', 'low insulin'],\n",
       " ['high pregnancies', 'low skin thickness', 'normal insulin'],\n",
       " ['high pregnancies', 'low skin thickness', 'high insulin'],\n",
       " ['high pregnancies', 'normal skin thickness', 'low insulin'],\n",
       " ['high pregnancies', 'normal skin thickness', 'normal insulin'],\n",
       " ['high pregnancies', 'normal skin thickness', 'high insulin'],\n",
       " ['high pregnancies', 'high skin thickness', 'low insulin'],\n",
       " ['high pregnancies', 'high skin thickness', 'normal insulin'],\n",
       " ['high pregnancies', 'high skin thickness', 'high insulin'],\n",
       " ['low pregnancies', 'low skin thickness', 'low bmi'],\n",
       " ['low pregnancies', 'low skin thickness', 'normal bmi'],\n",
       " ['low pregnancies', 'low skin thickness', 'high bmi'],\n",
       " ['low pregnancies', 'normal skin thickness', 'low bmi'],\n",
       " ['low pregnancies', 'normal skin thickness', 'normal bmi'],\n",
       " ['low pregnancies', 'normal skin thickness', 'high bmi'],\n",
       " ['low pregnancies', 'high skin thickness', 'low bmi'],\n",
       " ['low pregnancies', 'high skin thickness', 'normal bmi'],\n",
       " ['low pregnancies', 'high skin thickness', 'high bmi'],\n",
       " ['normal pregnancies', 'low skin thickness', 'low bmi'],\n",
       " ['normal pregnancies', 'low skin thickness', 'normal bmi'],\n",
       " ['normal pregnancies', 'low skin thickness', 'high bmi'],\n",
       " ['normal pregnancies', 'normal skin thickness', 'low bmi'],\n",
       " ['normal pregnancies', 'normal skin thickness', 'normal bmi'],\n",
       " ['normal pregnancies', 'normal skin thickness', 'high bmi'],\n",
       " ['normal pregnancies', 'high skin thickness', 'low bmi'],\n",
       " ['normal pregnancies', 'high skin thickness', 'normal bmi'],\n",
       " ['normal pregnancies', 'high skin thickness', 'high bmi'],\n",
       " ['high pregnancies', 'low skin thickness', 'low bmi'],\n",
       " ['high pregnancies', 'low skin thickness', 'normal bmi'],\n",
       " ['high pregnancies', 'low skin thickness', 'high bmi'],\n",
       " ['high pregnancies', 'normal skin thickness', 'low bmi'],\n",
       " ['high pregnancies', 'normal skin thickness', 'normal bmi'],\n",
       " ['high pregnancies', 'normal skin thickness', 'high bmi'],\n",
       " ['high pregnancies', 'high skin thickness', 'low bmi'],\n",
       " ['high pregnancies', 'high skin thickness', 'normal bmi'],\n",
       " ['high pregnancies', 'high skin thickness', 'high bmi'],\n",
       " ['low pregnancies', 'low skin thickness', 'low diabetes pedigree function'],\n",
       " ['low pregnancies',\n",
       "  'low skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'low skin thickness', 'high diabetes pedigree function'],\n",
       " ['low pregnancies',\n",
       "  'normal skin thickness',\n",
       "  'low diabetes pedigree function'],\n",
       " ['low pregnancies',\n",
       "  'normal skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['low pregnancies',\n",
       "  'normal skin thickness',\n",
       "  'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'high skin thickness', 'low diabetes pedigree function'],\n",
       " ['low pregnancies',\n",
       "  'high skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'high skin thickness', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'low skin thickness',\n",
       "  'low diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'low skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'low skin thickness',\n",
       "  'high diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'normal skin thickness',\n",
       "  'low diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'normal skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'normal skin thickness',\n",
       "  'high diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'high skin thickness',\n",
       "  'low diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'high skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['normal pregnancies',\n",
       "  'high skin thickness',\n",
       "  'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'low skin thickness', 'low diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'low skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'low skin thickness', 'high diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'normal skin thickness',\n",
       "  'low diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'normal skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'normal skin thickness',\n",
       "  'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'high skin thickness', 'low diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'high skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['high pregnancies',\n",
       "  'high skin thickness',\n",
       "  'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'low skin thickness', 'low age'],\n",
       " ['low pregnancies', 'low skin thickness', 'normal age'],\n",
       " ['low pregnancies', 'low skin thickness', 'high age'],\n",
       " ['low pregnancies', 'normal skin thickness', 'low age'],\n",
       " ['low pregnancies', 'normal skin thickness', 'normal age'],\n",
       " ['low pregnancies', 'normal skin thickness', 'high age'],\n",
       " ['low pregnancies', 'high skin thickness', 'low age'],\n",
       " ['low pregnancies', 'high skin thickness', 'normal age'],\n",
       " ['low pregnancies', 'high skin thickness', 'high age'],\n",
       " ['normal pregnancies', 'low skin thickness', 'low age'],\n",
       " ['normal pregnancies', 'low skin thickness', 'normal age'],\n",
       " ['normal pregnancies', 'low skin thickness', 'high age'],\n",
       " ['normal pregnancies', 'normal skin thickness', 'low age'],\n",
       " ['normal pregnancies', 'normal skin thickness', 'normal age'],\n",
       " ['normal pregnancies', 'normal skin thickness', 'high age'],\n",
       " ['normal pregnancies', 'high skin thickness', 'low age'],\n",
       " ['normal pregnancies', 'high skin thickness', 'normal age'],\n",
       " ['normal pregnancies', 'high skin thickness', 'high age'],\n",
       " ['high pregnancies', 'low skin thickness', 'low age'],\n",
       " ['high pregnancies', 'low skin thickness', 'normal age'],\n",
       " ['high pregnancies', 'low skin thickness', 'high age'],\n",
       " ['high pregnancies', 'normal skin thickness', 'low age'],\n",
       " ['high pregnancies', 'normal skin thickness', 'normal age'],\n",
       " ['high pregnancies', 'normal skin thickness', 'high age'],\n",
       " ['high pregnancies', 'high skin thickness', 'low age'],\n",
       " ['high pregnancies', 'high skin thickness', 'normal age'],\n",
       " ['high pregnancies', 'high skin thickness', 'high age'],\n",
       " ['low pregnancies', 'low insulin', 'low bmi'],\n",
       " ['low pregnancies', 'low insulin', 'normal bmi'],\n",
       " ['low pregnancies', 'low insulin', 'high bmi'],\n",
       " ['low pregnancies', 'normal insulin', 'low bmi'],\n",
       " ['low pregnancies', 'normal insulin', 'normal bmi'],\n",
       " ['low pregnancies', 'normal insulin', 'high bmi'],\n",
       " ['low pregnancies', 'high insulin', 'low bmi'],\n",
       " ['low pregnancies', 'high insulin', 'normal bmi'],\n",
       " ['low pregnancies', 'high insulin', 'high bmi'],\n",
       " ['normal pregnancies', 'low insulin', 'low bmi'],\n",
       " ['normal pregnancies', 'low insulin', 'normal bmi'],\n",
       " ['normal pregnancies', 'low insulin', 'high bmi'],\n",
       " ['normal pregnancies', 'normal insulin', 'low bmi'],\n",
       " ['normal pregnancies', 'normal insulin', 'normal bmi'],\n",
       " ['normal pregnancies', 'normal insulin', 'high bmi'],\n",
       " ['normal pregnancies', 'high insulin', 'low bmi'],\n",
       " ['normal pregnancies', 'high insulin', 'normal bmi'],\n",
       " ['normal pregnancies', 'high insulin', 'high bmi'],\n",
       " ['high pregnancies', 'low insulin', 'low bmi'],\n",
       " ['high pregnancies', 'low insulin', 'normal bmi'],\n",
       " ['high pregnancies', 'low insulin', 'high bmi'],\n",
       " ['high pregnancies', 'normal insulin', 'low bmi'],\n",
       " ['high pregnancies', 'normal insulin', 'normal bmi'],\n",
       " ['high pregnancies', 'normal insulin', 'high bmi'],\n",
       " ['high pregnancies', 'high insulin', 'low bmi'],\n",
       " ['high pregnancies', 'high insulin', 'normal bmi'],\n",
       " ['high pregnancies', 'high insulin', 'high bmi'],\n",
       " ['low pregnancies', 'low insulin', 'low diabetes pedigree function'],\n",
       " ['low pregnancies', 'low insulin', 'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'low insulin', 'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'normal insulin', 'low diabetes pedigree function'],\n",
       " ['low pregnancies', 'normal insulin', 'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'normal insulin', 'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'high insulin', 'low diabetes pedigree function'],\n",
       " ['low pregnancies', 'high insulin', 'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'high insulin', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies', 'low insulin', 'low diabetes pedigree function'],\n",
       " ['normal pregnancies', 'low insulin', 'normal diabetes pedigree function'],\n",
       " ['normal pregnancies', 'low insulin', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies', 'normal insulin', 'low diabetes pedigree function'],\n",
       " ['normal pregnancies', 'normal insulin', 'normal diabetes pedigree function'],\n",
       " ['normal pregnancies', 'normal insulin', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies', 'high insulin', 'low diabetes pedigree function'],\n",
       " ['normal pregnancies', 'high insulin', 'normal diabetes pedigree function'],\n",
       " ['normal pregnancies', 'high insulin', 'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'low insulin', 'low diabetes pedigree function'],\n",
       " ['high pregnancies', 'low insulin', 'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'low insulin', 'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'normal insulin', 'low diabetes pedigree function'],\n",
       " ['high pregnancies', 'normal insulin', 'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'normal insulin', 'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'high insulin', 'low diabetes pedigree function'],\n",
       " ['high pregnancies', 'high insulin', 'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'high insulin', 'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'low insulin', 'low age'],\n",
       " ['low pregnancies', 'low insulin', 'normal age'],\n",
       " ['low pregnancies', 'low insulin', 'high age'],\n",
       " ['low pregnancies', 'normal insulin', 'low age'],\n",
       " ['low pregnancies', 'normal insulin', 'normal age'],\n",
       " ['low pregnancies', 'normal insulin', 'high age'],\n",
       " ['low pregnancies', 'high insulin', 'low age'],\n",
       " ['low pregnancies', 'high insulin', 'normal age'],\n",
       " ['low pregnancies', 'high insulin', 'high age'],\n",
       " ['normal pregnancies', 'low insulin', 'low age'],\n",
       " ['normal pregnancies', 'low insulin', 'normal age'],\n",
       " ['normal pregnancies', 'low insulin', 'high age'],\n",
       " ['normal pregnancies', 'normal insulin', 'low age'],\n",
       " ['normal pregnancies', 'normal insulin', 'normal age'],\n",
       " ['normal pregnancies', 'normal insulin', 'high age'],\n",
       " ['normal pregnancies', 'high insulin', 'low age'],\n",
       " ['normal pregnancies', 'high insulin', 'normal age'],\n",
       " ['normal pregnancies', 'high insulin', 'high age'],\n",
       " ['high pregnancies', 'low insulin', 'low age'],\n",
       " ['high pregnancies', 'low insulin', 'normal age'],\n",
       " ['high pregnancies', 'low insulin', 'high age'],\n",
       " ['high pregnancies', 'normal insulin', 'low age'],\n",
       " ['high pregnancies', 'normal insulin', 'normal age'],\n",
       " ['high pregnancies', 'normal insulin', 'high age'],\n",
       " ['high pregnancies', 'high insulin', 'low age'],\n",
       " ['high pregnancies', 'high insulin', 'normal age'],\n",
       " ['high pregnancies', 'high insulin', 'high age'],\n",
       " ['low pregnancies', 'low bmi', 'low diabetes pedigree function'],\n",
       " ['low pregnancies', 'low bmi', 'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'low bmi', 'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'normal bmi', 'low diabetes pedigree function'],\n",
       " ['low pregnancies', 'normal bmi', 'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'normal bmi', 'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'high bmi', 'low diabetes pedigree function'],\n",
       " ['low pregnancies', 'high bmi', 'normal diabetes pedigree function'],\n",
       " ['low pregnancies', 'high bmi', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies', 'low bmi', 'low diabetes pedigree function'],\n",
       " ['normal pregnancies', 'low bmi', 'normal diabetes pedigree function'],\n",
       " ['normal pregnancies', 'low bmi', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies', 'normal bmi', 'low diabetes pedigree function'],\n",
       " ['normal pregnancies', 'normal bmi', 'normal diabetes pedigree function'],\n",
       " ['normal pregnancies', 'normal bmi', 'high diabetes pedigree function'],\n",
       " ['normal pregnancies', 'high bmi', 'low diabetes pedigree function'],\n",
       " ['normal pregnancies', 'high bmi', 'normal diabetes pedigree function'],\n",
       " ['normal pregnancies', 'high bmi', 'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'low bmi', 'low diabetes pedigree function'],\n",
       " ['high pregnancies', 'low bmi', 'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'low bmi', 'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'normal bmi', 'low diabetes pedigree function'],\n",
       " ['high pregnancies', 'normal bmi', 'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'normal bmi', 'high diabetes pedigree function'],\n",
       " ['high pregnancies', 'high bmi', 'low diabetes pedigree function'],\n",
       " ['high pregnancies', 'high bmi', 'normal diabetes pedigree function'],\n",
       " ['high pregnancies', 'high bmi', 'high diabetes pedigree function'],\n",
       " ['low pregnancies', 'low bmi', 'low age'],\n",
       " ['low pregnancies', 'low bmi', 'normal age'],\n",
       " ['low pregnancies', 'low bmi', 'high age'],\n",
       " ['low pregnancies', 'normal bmi', 'low age'],\n",
       " ['low pregnancies', 'normal bmi', 'normal age'],\n",
       " ['low pregnancies', 'normal bmi', 'high age'],\n",
       " ['low pregnancies', 'high bmi', 'low age'],\n",
       " ['low pregnancies', 'high bmi', 'normal age'],\n",
       " ['low pregnancies', 'high bmi', 'high age'],\n",
       " ['normal pregnancies', 'low bmi', 'low age'],\n",
       " ['normal pregnancies', 'low bmi', 'normal age'],\n",
       " ['normal pregnancies', 'low bmi', 'high age'],\n",
       " ['normal pregnancies', 'normal bmi', 'low age'],\n",
       " ['normal pregnancies', 'normal bmi', 'normal age'],\n",
       " ['normal pregnancies', 'normal bmi', 'high age'],\n",
       " ['normal pregnancies', 'high bmi', 'low age'],\n",
       " ['normal pregnancies', 'high bmi', 'normal age'],\n",
       " ['normal pregnancies', 'high bmi', 'high age'],\n",
       " ['high pregnancies', 'low bmi', 'low age'],\n",
       " ['high pregnancies', 'low bmi', 'normal age'],\n",
       " ['high pregnancies', 'low bmi', 'high age'],\n",
       " ['high pregnancies', 'normal bmi', 'low age'],\n",
       " ['high pregnancies', 'normal bmi', 'normal age'],\n",
       " ['high pregnancies', 'normal bmi', 'high age'],\n",
       " ['high pregnancies', 'high bmi', 'low age'],\n",
       " ['high pregnancies', 'high bmi', 'normal age'],\n",
       " ['high pregnancies', 'high bmi', 'high age'],\n",
       " ['low pregnancies', 'low diabetes pedigree function', 'low age'],\n",
       " ['low pregnancies', 'low diabetes pedigree function', 'normal age'],\n",
       " ['low pregnancies', 'low diabetes pedigree function', 'high age'],\n",
       " ['low pregnancies', 'normal diabetes pedigree function', 'low age'],\n",
       " ['low pregnancies', 'normal diabetes pedigree function', 'normal age'],\n",
       " ['low pregnancies', 'normal diabetes pedigree function', 'high age'],\n",
       " ['low pregnancies', 'high diabetes pedigree function', 'low age'],\n",
       " ['low pregnancies', 'high diabetes pedigree function', 'normal age'],\n",
       " ['low pregnancies', 'high diabetes pedigree function', 'high age'],\n",
       " ['normal pregnancies', 'low diabetes pedigree function', 'low age'],\n",
       " ['normal pregnancies', 'low diabetes pedigree function', 'normal age'],\n",
       " ['normal pregnancies', 'low diabetes pedigree function', 'high age'],\n",
       " ['normal pregnancies', 'normal diabetes pedigree function', 'low age'],\n",
       " ['normal pregnancies', 'normal diabetes pedigree function', 'normal age'],\n",
       " ['normal pregnancies', 'normal diabetes pedigree function', 'high age'],\n",
       " ['normal pregnancies', 'high diabetes pedigree function', 'low age'],\n",
       " ['normal pregnancies', 'high diabetes pedigree function', 'normal age'],\n",
       " ['normal pregnancies', 'high diabetes pedigree function', 'high age'],\n",
       " ['high pregnancies', 'low diabetes pedigree function', 'low age'],\n",
       " ['high pregnancies', 'low diabetes pedigree function', 'normal age'],\n",
       " ['high pregnancies', 'low diabetes pedigree function', 'high age'],\n",
       " ['high pregnancies', 'normal diabetes pedigree function', 'low age'],\n",
       " ['high pregnancies', 'normal diabetes pedigree function', 'normal age'],\n",
       " ['high pregnancies', 'normal diabetes pedigree function', 'high age'],\n",
       " ['high pregnancies', 'high diabetes pedigree function', 'low age'],\n",
       " ['high pregnancies', 'high diabetes pedigree function', 'normal age'],\n",
       " ['high pregnancies', 'high diabetes pedigree function', 'high age'],\n",
       " ['low glucose', 'low blood pressure', 'low skin thickness'],\n",
       " ['low glucose', 'low blood pressure', 'normal skin thickness'],\n",
       " ['low glucose', 'low blood pressure', 'high skin thickness'],\n",
       " ['low glucose', 'normal blood pressure', 'low skin thickness'],\n",
       " ['low glucose', 'normal blood pressure', 'normal skin thickness'],\n",
       " ['low glucose', 'normal blood pressure', 'high skin thickness'],\n",
       " ['low glucose', 'high blood pressure', 'low skin thickness'],\n",
       " ['low glucose', 'high blood pressure', 'normal skin thickness'],\n",
       " ['low glucose', 'high blood pressure', 'high skin thickness'],\n",
       " ['normal glucose', 'low blood pressure', 'low skin thickness'],\n",
       " ['normal glucose', 'low blood pressure', 'normal skin thickness'],\n",
       " ['normal glucose', 'low blood pressure', 'high skin thickness'],\n",
       " ['normal glucose', 'normal blood pressure', 'low skin thickness'],\n",
       " ['normal glucose', 'normal blood pressure', 'normal skin thickness'],\n",
       " ['normal glucose', 'normal blood pressure', 'high skin thickness'],\n",
       " ['normal glucose', 'high blood pressure', 'low skin thickness'],\n",
       " ['normal glucose', 'high blood pressure', 'normal skin thickness'],\n",
       " ['normal glucose', 'high blood pressure', 'high skin thickness'],\n",
       " ['high glucose', 'low blood pressure', 'low skin thickness'],\n",
       " ['high glucose', 'low blood pressure', 'normal skin thickness'],\n",
       " ['high glucose', 'low blood pressure', 'high skin thickness'],\n",
       " ['high glucose', 'normal blood pressure', 'low skin thickness'],\n",
       " ['high glucose', 'normal blood pressure', 'normal skin thickness'],\n",
       " ['high glucose', 'normal blood pressure', 'high skin thickness'],\n",
       " ['high glucose', 'high blood pressure', 'low skin thickness'],\n",
       " ['high glucose', 'high blood pressure', 'normal skin thickness'],\n",
       " ['high glucose', 'high blood pressure', 'high skin thickness'],\n",
       " ['low glucose', 'low blood pressure', 'low insulin'],\n",
       " ['low glucose', 'low blood pressure', 'normal insulin'],\n",
       " ['low glucose', 'low blood pressure', 'high insulin'],\n",
       " ['low glucose', 'normal blood pressure', 'low insulin'],\n",
       " ['low glucose', 'normal blood pressure', 'normal insulin'],\n",
       " ['low glucose', 'normal blood pressure', 'high insulin'],\n",
       " ['low glucose', 'high blood pressure', 'low insulin'],\n",
       " ['low glucose', 'high blood pressure', 'normal insulin'],\n",
       " ['low glucose', 'high blood pressure', 'high insulin'],\n",
       " ['normal glucose', 'low blood pressure', 'low insulin'],\n",
       " ['normal glucose', 'low blood pressure', 'normal insulin'],\n",
       " ['normal glucose', 'low blood pressure', 'high insulin'],\n",
       " ['normal glucose', 'normal blood pressure', 'low insulin'],\n",
       " ['normal glucose', 'normal blood pressure', 'normal insulin'],\n",
       " ['normal glucose', 'normal blood pressure', 'high insulin'],\n",
       " ['normal glucose', 'high blood pressure', 'low insulin'],\n",
       " ['normal glucose', 'high blood pressure', 'normal insulin'],\n",
       " ['normal glucose', 'high blood pressure', 'high insulin'],\n",
       " ['high glucose', 'low blood pressure', 'low insulin'],\n",
       " ['high glucose', 'low blood pressure', 'normal insulin'],\n",
       " ['high glucose', 'low blood pressure', 'high insulin'],\n",
       " ['high glucose', 'normal blood pressure', 'low insulin'],\n",
       " ['high glucose', 'normal blood pressure', 'normal insulin'],\n",
       " ['high glucose', 'normal blood pressure', 'high insulin'],\n",
       " ['high glucose', 'high blood pressure', 'low insulin'],\n",
       " ['high glucose', 'high blood pressure', 'normal insulin'],\n",
       " ['high glucose', 'high blood pressure', 'high insulin'],\n",
       " ['low glucose', 'low blood pressure', 'low bmi'],\n",
       " ['low glucose', 'low blood pressure', 'normal bmi'],\n",
       " ['low glucose', 'low blood pressure', 'high bmi'],\n",
       " ['low glucose', 'normal blood pressure', 'low bmi'],\n",
       " ['low glucose', 'normal blood pressure', 'normal bmi'],\n",
       " ['low glucose', 'normal blood pressure', 'high bmi'],\n",
       " ['low glucose', 'high blood pressure', 'low bmi'],\n",
       " ['low glucose', 'high blood pressure', 'normal bmi'],\n",
       " ['low glucose', 'high blood pressure', 'high bmi'],\n",
       " ['normal glucose', 'low blood pressure', 'low bmi'],\n",
       " ['normal glucose', 'low blood pressure', 'normal bmi'],\n",
       " ['normal glucose', 'low blood pressure', 'high bmi'],\n",
       " ['normal glucose', 'normal blood pressure', 'low bmi'],\n",
       " ['normal glucose', 'normal blood pressure', 'normal bmi'],\n",
       " ['normal glucose', 'normal blood pressure', 'high bmi'],\n",
       " ['normal glucose', 'high blood pressure', 'low bmi'],\n",
       " ['normal glucose', 'high blood pressure', 'normal bmi'],\n",
       " ['normal glucose', 'high blood pressure', 'high bmi'],\n",
       " ['high glucose', 'low blood pressure', 'low bmi'],\n",
       " ['high glucose', 'low blood pressure', 'normal bmi'],\n",
       " ['high glucose', 'low blood pressure', 'high bmi'],\n",
       " ['high glucose', 'normal blood pressure', 'low bmi'],\n",
       " ['high glucose', 'normal blood pressure', 'normal bmi'],\n",
       " ['high glucose', 'normal blood pressure', 'high bmi'],\n",
       " ['high glucose', 'high blood pressure', 'low bmi'],\n",
       " ['high glucose', 'high blood pressure', 'normal bmi'],\n",
       " ['high glucose', 'high blood pressure', 'high bmi'],\n",
       " ['low glucose', 'low blood pressure', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'low blood pressure', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'low blood pressure', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'normal blood pressure', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'normal blood pressure', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'normal blood pressure', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'high blood pressure', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'high blood pressure', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'high blood pressure', 'high diabetes pedigree function'],\n",
       " ['normal glucose', 'low blood pressure', 'low diabetes pedigree function'],\n",
       " ['normal glucose', 'low blood pressure', 'normal diabetes pedigree function'],\n",
       " ['normal glucose', 'low blood pressure', 'high diabetes pedigree function'],\n",
       " ['normal glucose', 'normal blood pressure', 'low diabetes pedigree function'],\n",
       " ['normal glucose',\n",
       "  'normal blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['normal glucose',\n",
       "  'normal blood pressure',\n",
       "  'high diabetes pedigree function'],\n",
       " ['normal glucose', 'high blood pressure', 'low diabetes pedigree function'],\n",
       " ['normal glucose',\n",
       "  'high blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['normal glucose', 'high blood pressure', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'low blood pressure', 'low diabetes pedigree function'],\n",
       " ['high glucose', 'low blood pressure', 'normal diabetes pedigree function'],\n",
       " ['high glucose', 'low blood pressure', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'normal blood pressure', 'low diabetes pedigree function'],\n",
       " ['high glucose',\n",
       "  'normal blood pressure',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['high glucose', 'normal blood pressure', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'high blood pressure', 'low diabetes pedigree function'],\n",
       " ['high glucose', 'high blood pressure', 'normal diabetes pedigree function'],\n",
       " ['high glucose', 'high blood pressure', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'low blood pressure', 'low age'],\n",
       " ['low glucose', 'low blood pressure', 'normal age'],\n",
       " ['low glucose', 'low blood pressure', 'high age'],\n",
       " ['low glucose', 'normal blood pressure', 'low age'],\n",
       " ['low glucose', 'normal blood pressure', 'normal age'],\n",
       " ['low glucose', 'normal blood pressure', 'high age'],\n",
       " ['low glucose', 'high blood pressure', 'low age'],\n",
       " ['low glucose', 'high blood pressure', 'normal age'],\n",
       " ['low glucose', 'high blood pressure', 'high age'],\n",
       " ['normal glucose', 'low blood pressure', 'low age'],\n",
       " ['normal glucose', 'low blood pressure', 'normal age'],\n",
       " ['normal glucose', 'low blood pressure', 'high age'],\n",
       " ['normal glucose', 'normal blood pressure', 'low age'],\n",
       " ['normal glucose', 'normal blood pressure', 'normal age'],\n",
       " ['normal glucose', 'normal blood pressure', 'high age'],\n",
       " ['normal glucose', 'high blood pressure', 'low age'],\n",
       " ['normal glucose', 'high blood pressure', 'normal age'],\n",
       " ['normal glucose', 'high blood pressure', 'high age'],\n",
       " ['high glucose', 'low blood pressure', 'low age'],\n",
       " ['high glucose', 'low blood pressure', 'normal age'],\n",
       " ['high glucose', 'low blood pressure', 'high age'],\n",
       " ['high glucose', 'normal blood pressure', 'low age'],\n",
       " ['high glucose', 'normal blood pressure', 'normal age'],\n",
       " ['high glucose', 'normal blood pressure', 'high age'],\n",
       " ['high glucose', 'high blood pressure', 'low age'],\n",
       " ['high glucose', 'high blood pressure', 'normal age'],\n",
       " ['high glucose', 'high blood pressure', 'high age'],\n",
       " ['low glucose', 'low skin thickness', 'low insulin'],\n",
       " ['low glucose', 'low skin thickness', 'normal insulin'],\n",
       " ['low glucose', 'low skin thickness', 'high insulin'],\n",
       " ['low glucose', 'normal skin thickness', 'low insulin'],\n",
       " ['low glucose', 'normal skin thickness', 'normal insulin'],\n",
       " ['low glucose', 'normal skin thickness', 'high insulin'],\n",
       " ['low glucose', 'high skin thickness', 'low insulin'],\n",
       " ['low glucose', 'high skin thickness', 'normal insulin'],\n",
       " ['low glucose', 'high skin thickness', 'high insulin'],\n",
       " ['normal glucose', 'low skin thickness', 'low insulin'],\n",
       " ['normal glucose', 'low skin thickness', 'normal insulin'],\n",
       " ['normal glucose', 'low skin thickness', 'high insulin'],\n",
       " ['normal glucose', 'normal skin thickness', 'low insulin'],\n",
       " ['normal glucose', 'normal skin thickness', 'normal insulin'],\n",
       " ['normal glucose', 'normal skin thickness', 'high insulin'],\n",
       " ['normal glucose', 'high skin thickness', 'low insulin'],\n",
       " ['normal glucose', 'high skin thickness', 'normal insulin'],\n",
       " ['normal glucose', 'high skin thickness', 'high insulin'],\n",
       " ['high glucose', 'low skin thickness', 'low insulin'],\n",
       " ['high glucose', 'low skin thickness', 'normal insulin'],\n",
       " ['high glucose', 'low skin thickness', 'high insulin'],\n",
       " ['high glucose', 'normal skin thickness', 'low insulin'],\n",
       " ['high glucose', 'normal skin thickness', 'normal insulin'],\n",
       " ['high glucose', 'normal skin thickness', 'high insulin'],\n",
       " ['high glucose', 'high skin thickness', 'low insulin'],\n",
       " ['high glucose', 'high skin thickness', 'normal insulin'],\n",
       " ['high glucose', 'high skin thickness', 'high insulin'],\n",
       " ['low glucose', 'low skin thickness', 'low bmi'],\n",
       " ['low glucose', 'low skin thickness', 'normal bmi'],\n",
       " ['low glucose', 'low skin thickness', 'high bmi'],\n",
       " ['low glucose', 'normal skin thickness', 'low bmi'],\n",
       " ['low glucose', 'normal skin thickness', 'normal bmi'],\n",
       " ['low glucose', 'normal skin thickness', 'high bmi'],\n",
       " ['low glucose', 'high skin thickness', 'low bmi'],\n",
       " ['low glucose', 'high skin thickness', 'normal bmi'],\n",
       " ['low glucose', 'high skin thickness', 'high bmi'],\n",
       " ['normal glucose', 'low skin thickness', 'low bmi'],\n",
       " ['normal glucose', 'low skin thickness', 'normal bmi'],\n",
       " ['normal glucose', 'low skin thickness', 'high bmi'],\n",
       " ['normal glucose', 'normal skin thickness', 'low bmi'],\n",
       " ['normal glucose', 'normal skin thickness', 'normal bmi'],\n",
       " ['normal glucose', 'normal skin thickness', 'high bmi'],\n",
       " ['normal glucose', 'high skin thickness', 'low bmi'],\n",
       " ['normal glucose', 'high skin thickness', 'normal bmi'],\n",
       " ['normal glucose', 'high skin thickness', 'high bmi'],\n",
       " ['high glucose', 'low skin thickness', 'low bmi'],\n",
       " ['high glucose', 'low skin thickness', 'normal bmi'],\n",
       " ['high glucose', 'low skin thickness', 'high bmi'],\n",
       " ['high glucose', 'normal skin thickness', 'low bmi'],\n",
       " ['high glucose', 'normal skin thickness', 'normal bmi'],\n",
       " ['high glucose', 'normal skin thickness', 'high bmi'],\n",
       " ['high glucose', 'high skin thickness', 'low bmi'],\n",
       " ['high glucose', 'high skin thickness', 'normal bmi'],\n",
       " ['high glucose', 'high skin thickness', 'high bmi'],\n",
       " ['low glucose', 'low skin thickness', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'low skin thickness', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'low skin thickness', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'normal skin thickness', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'normal skin thickness', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'normal skin thickness', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'high skin thickness', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'high skin thickness', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'high skin thickness', 'high diabetes pedigree function'],\n",
       " ['normal glucose', 'low skin thickness', 'low diabetes pedigree function'],\n",
       " ['normal glucose', 'low skin thickness', 'normal diabetes pedigree function'],\n",
       " ['normal glucose', 'low skin thickness', 'high diabetes pedigree function'],\n",
       " ['normal glucose', 'normal skin thickness', 'low diabetes pedigree function'],\n",
       " ['normal glucose',\n",
       "  'normal skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['normal glucose',\n",
       "  'normal skin thickness',\n",
       "  'high diabetes pedigree function'],\n",
       " ['normal glucose', 'high skin thickness', 'low diabetes pedigree function'],\n",
       " ['normal glucose',\n",
       "  'high skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['normal glucose', 'high skin thickness', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'low skin thickness', 'low diabetes pedigree function'],\n",
       " ['high glucose', 'low skin thickness', 'normal diabetes pedigree function'],\n",
       " ['high glucose', 'low skin thickness', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'normal skin thickness', 'low diabetes pedigree function'],\n",
       " ['high glucose',\n",
       "  'normal skin thickness',\n",
       "  'normal diabetes pedigree function'],\n",
       " ['high glucose', 'normal skin thickness', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'high skin thickness', 'low diabetes pedigree function'],\n",
       " ['high glucose', 'high skin thickness', 'normal diabetes pedigree function'],\n",
       " ['high glucose', 'high skin thickness', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'low skin thickness', 'low age'],\n",
       " ['low glucose', 'low skin thickness', 'normal age'],\n",
       " ['low glucose', 'low skin thickness', 'high age'],\n",
       " ['low glucose', 'normal skin thickness', 'low age'],\n",
       " ['low glucose', 'normal skin thickness', 'normal age'],\n",
       " ['low glucose', 'normal skin thickness', 'high age'],\n",
       " ['low glucose', 'high skin thickness', 'low age'],\n",
       " ['low glucose', 'high skin thickness', 'normal age'],\n",
       " ['low glucose', 'high skin thickness', 'high age'],\n",
       " ['normal glucose', 'low skin thickness', 'low age'],\n",
       " ['normal glucose', 'low skin thickness', 'normal age'],\n",
       " ['normal glucose', 'low skin thickness', 'high age'],\n",
       " ['normal glucose', 'normal skin thickness', 'low age'],\n",
       " ['normal glucose', 'normal skin thickness', 'normal age'],\n",
       " ['normal glucose', 'normal skin thickness', 'high age'],\n",
       " ['normal glucose', 'high skin thickness', 'low age'],\n",
       " ['normal glucose', 'high skin thickness', 'normal age'],\n",
       " ['normal glucose', 'high skin thickness', 'high age'],\n",
       " ['high glucose', 'low skin thickness', 'low age'],\n",
       " ['high glucose', 'low skin thickness', 'normal age'],\n",
       " ['high glucose', 'low skin thickness', 'high age'],\n",
       " ['high glucose', 'normal skin thickness', 'low age'],\n",
       " ['high glucose', 'normal skin thickness', 'normal age'],\n",
       " ['high glucose', 'normal skin thickness', 'high age'],\n",
       " ['high glucose', 'high skin thickness', 'low age'],\n",
       " ['high glucose', 'high skin thickness', 'normal age'],\n",
       " ['high glucose', 'high skin thickness', 'high age'],\n",
       " ['low glucose', 'low insulin', 'low bmi'],\n",
       " ['low glucose', 'low insulin', 'normal bmi'],\n",
       " ['low glucose', 'low insulin', 'high bmi'],\n",
       " ['low glucose', 'normal insulin', 'low bmi'],\n",
       " ['low glucose', 'normal insulin', 'normal bmi'],\n",
       " ['low glucose', 'normal insulin', 'high bmi'],\n",
       " ['low glucose', 'high insulin', 'low bmi'],\n",
       " ['low glucose', 'high insulin', 'normal bmi'],\n",
       " ['low glucose', 'high insulin', 'high bmi'],\n",
       " ['normal glucose', 'low insulin', 'low bmi'],\n",
       " ['normal glucose', 'low insulin', 'normal bmi'],\n",
       " ['normal glucose', 'low insulin', 'high bmi'],\n",
       " ['normal glucose', 'normal insulin', 'low bmi'],\n",
       " ['normal glucose', 'normal insulin', 'normal bmi'],\n",
       " ['normal glucose', 'normal insulin', 'high bmi'],\n",
       " ['normal glucose', 'high insulin', 'low bmi'],\n",
       " ['normal glucose', 'high insulin', 'normal bmi'],\n",
       " ['normal glucose', 'high insulin', 'high bmi'],\n",
       " ['high glucose', 'low insulin', 'low bmi'],\n",
       " ['high glucose', 'low insulin', 'normal bmi'],\n",
       " ['high glucose', 'low insulin', 'high bmi'],\n",
       " ['high glucose', 'normal insulin', 'low bmi'],\n",
       " ['high glucose', 'normal insulin', 'normal bmi'],\n",
       " ['high glucose', 'normal insulin', 'high bmi'],\n",
       " ['high glucose', 'high insulin', 'low bmi'],\n",
       " ['high glucose', 'high insulin', 'normal bmi'],\n",
       " ['high glucose', 'high insulin', 'high bmi'],\n",
       " ['low glucose', 'low insulin', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'low insulin', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'low insulin', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'normal insulin', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'normal insulin', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'normal insulin', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'high insulin', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'high insulin', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'high insulin', 'high diabetes pedigree function'],\n",
       " ['normal glucose', 'low insulin', 'low diabetes pedigree function'],\n",
       " ['normal glucose', 'low insulin', 'normal diabetes pedigree function'],\n",
       " ['normal glucose', 'low insulin', 'high diabetes pedigree function'],\n",
       " ['normal glucose', 'normal insulin', 'low diabetes pedigree function'],\n",
       " ['normal glucose', 'normal insulin', 'normal diabetes pedigree function'],\n",
       " ['normal glucose', 'normal insulin', 'high diabetes pedigree function'],\n",
       " ['normal glucose', 'high insulin', 'low diabetes pedigree function'],\n",
       " ['normal glucose', 'high insulin', 'normal diabetes pedigree function'],\n",
       " ['normal glucose', 'high insulin', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'low insulin', 'low diabetes pedigree function'],\n",
       " ['high glucose', 'low insulin', 'normal diabetes pedigree function'],\n",
       " ['high glucose', 'low insulin', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'normal insulin', 'low diabetes pedigree function'],\n",
       " ['high glucose', 'normal insulin', 'normal diabetes pedigree function'],\n",
       " ['high glucose', 'normal insulin', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'high insulin', 'low diabetes pedigree function'],\n",
       " ['high glucose', 'high insulin', 'normal diabetes pedigree function'],\n",
       " ['high glucose', 'high insulin', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'low insulin', 'low age'],\n",
       " ['low glucose', 'low insulin', 'normal age'],\n",
       " ['low glucose', 'low insulin', 'high age'],\n",
       " ['low glucose', 'normal insulin', 'low age'],\n",
       " ['low glucose', 'normal insulin', 'normal age'],\n",
       " ['low glucose', 'normal insulin', 'high age'],\n",
       " ['low glucose', 'high insulin', 'low age'],\n",
       " ['low glucose', 'high insulin', 'normal age'],\n",
       " ['low glucose', 'high insulin', 'high age'],\n",
       " ['normal glucose', 'low insulin', 'low age'],\n",
       " ['normal glucose', 'low insulin', 'normal age'],\n",
       " ['normal glucose', 'low insulin', 'high age'],\n",
       " ['normal glucose', 'normal insulin', 'low age'],\n",
       " ['normal glucose', 'normal insulin', 'normal age'],\n",
       " ['normal glucose', 'normal insulin', 'high age'],\n",
       " ['normal glucose', 'high insulin', 'low age'],\n",
       " ['normal glucose', 'high insulin', 'normal age'],\n",
       " ['normal glucose', 'high insulin', 'high age'],\n",
       " ['high glucose', 'low insulin', 'low age'],\n",
       " ['high glucose', 'low insulin', 'normal age'],\n",
       " ['high glucose', 'low insulin', 'high age'],\n",
       " ['high glucose', 'normal insulin', 'low age'],\n",
       " ['high glucose', 'normal insulin', 'normal age'],\n",
       " ['high glucose', 'normal insulin', 'high age'],\n",
       " ['high glucose', 'high insulin', 'low age'],\n",
       " ['high glucose', 'high insulin', 'normal age'],\n",
       " ['high glucose', 'high insulin', 'high age'],\n",
       " ['low glucose', 'low bmi', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'low bmi', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'low bmi', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'normal bmi', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'normal bmi', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'normal bmi', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'high bmi', 'low diabetes pedigree function'],\n",
       " ['low glucose', 'high bmi', 'normal diabetes pedigree function'],\n",
       " ['low glucose', 'high bmi', 'high diabetes pedigree function'],\n",
       " ['normal glucose', 'low bmi', 'low diabetes pedigree function'],\n",
       " ['normal glucose', 'low bmi', 'normal diabetes pedigree function'],\n",
       " ['normal glucose', 'low bmi', 'high diabetes pedigree function'],\n",
       " ['normal glucose', 'normal bmi', 'low diabetes pedigree function'],\n",
       " ['normal glucose', 'normal bmi', 'normal diabetes pedigree function'],\n",
       " ['normal glucose', 'normal bmi', 'high diabetes pedigree function'],\n",
       " ['normal glucose', 'high bmi', 'low diabetes pedigree function'],\n",
       " ['normal glucose', 'high bmi', 'normal diabetes pedigree function'],\n",
       " ['normal glucose', 'high bmi', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'low bmi', 'low diabetes pedigree function'],\n",
       " ['high glucose', 'low bmi', 'normal diabetes pedigree function'],\n",
       " ['high glucose', 'low bmi', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'normal bmi', 'low diabetes pedigree function'],\n",
       " ['high glucose', 'normal bmi', 'normal diabetes pedigree function'],\n",
       " ['high glucose', 'normal bmi', 'high diabetes pedigree function'],\n",
       " ['high glucose', 'high bmi', 'low diabetes pedigree function'],\n",
       " ['high glucose', 'high bmi', 'normal diabetes pedigree function'],\n",
       " ['high glucose', 'high bmi', 'high diabetes pedigree function'],\n",
       " ['low glucose', 'low bmi', 'low age'],\n",
       " ['low glucose', 'low bmi', 'normal age'],\n",
       " ['low glucose', 'low bmi', 'high age'],\n",
       " ['low glucose', 'normal bmi', 'low age'],\n",
       " ['low glucose', 'normal bmi', 'normal age'],\n",
       " ['low glucose', 'normal bmi', 'high age'],\n",
       " ['low glucose', 'high bmi', 'low age'],\n",
       " ['low glucose', 'high bmi', 'normal age'],\n",
       " ['low glucose', 'high bmi', 'high age'],\n",
       " ['normal glucose', 'low bmi', 'low age'],\n",
       " ['normal glucose', 'low bmi', 'normal age'],\n",
       " ['normal glucose', 'low bmi', 'high age'],\n",
       " ['normal glucose', 'normal bmi', 'low age'],\n",
       " ['normal glucose', 'normal bmi', 'normal age'],\n",
       " ['normal glucose', 'normal bmi', 'high age'],\n",
       " ['normal glucose', 'high bmi', 'low age'],\n",
       " ['normal glucose', 'high bmi', 'normal age'],\n",
       " ['normal glucose', 'high bmi', 'high age'],\n",
       " ['high glucose', 'low bmi', 'low age'],\n",
       " ['high glucose', 'low bmi', 'normal age'],\n",
       " ['high glucose', 'low bmi', 'high age'],\n",
       " ['high glucose', 'normal bmi', 'low age'],\n",
       " ['high glucose', 'normal bmi', 'normal age'],\n",
       " ['high glucose', 'normal bmi', 'high age'],\n",
       " ['high glucose', 'high bmi', 'low age'],\n",
       " ['high glucose', 'high bmi', 'normal age'],\n",
       " ['high glucose', 'high bmi', 'high age'],\n",
       " ['low glucose', 'low diabetes pedigree function', 'low age'],\n",
       " ['low glucose', 'low diabetes pedigree function', 'normal age'],\n",
       " ['low glucose', 'low diabetes pedigree function', 'high age'],\n",
       " ['low glucose', 'normal diabetes pedigree function', 'low age'],\n",
       " ['low glucose', 'normal diabetes pedigree function', 'normal age'],\n",
       " ['low glucose', 'normal diabetes pedigree function', 'high age'],\n",
       " ['low glucose', 'high diabetes pedigree function', 'low age'],\n",
       " ['low glucose', 'high diabetes pedigree function', 'normal age'],\n",
       " ['low glucose', 'high diabetes pedigree function', 'high age'],\n",
       " ['normal glucose', 'low diabetes pedigree function', 'low age'],\n",
       " ['normal glucose', 'low diabetes pedigree function', 'normal age'],\n",
       " ['normal glucose', 'low diabetes pedigree function', 'high age'],\n",
       " ['normal glucose', 'normal diabetes pedigree function', 'low age'],\n",
       " ['normal glucose', 'normal diabetes pedigree function', 'normal age'],\n",
       " ['normal glucose', 'normal diabetes pedigree function', 'high age'],\n",
       " ['normal glucose', 'high diabetes pedigree function', 'low age'],\n",
       " ['normal glucose', 'high diabetes pedigree function', 'normal age'],\n",
       " ['normal glucose', 'high diabetes pedigree function', 'high age'],\n",
       " ['high glucose', 'low diabetes pedigree function', 'low age'],\n",
       " ['high glucose', 'low diabetes pedigree function', 'normal age'],\n",
       " ['high glucose', 'low diabetes pedigree function', 'high age'],\n",
       " ['high glucose', 'normal diabetes pedigree function', 'low age'],\n",
       " ['high glucose', 'normal diabetes pedigree function', 'normal age'],\n",
       " ['high glucose', 'normal diabetes pedigree function', 'high age'],\n",
       " ['high glucose', 'high diabetes pedigree function', 'low age'],\n",
       " ['high glucose', 'high diabetes pedigree function', 'normal age'],\n",
       " ['high glucose', 'high diabetes pedigree function', 'high age'],\n",
       " ['low blood pressure', 'low skin thickness', 'low insulin'],\n",
       " ['low blood pressure', 'low skin thickness', 'normal insulin'],\n",
       " ['low blood pressure', 'low skin thickness', 'high insulin'],\n",
       " ['low blood pressure', 'normal skin thickness', 'low insulin'],\n",
       " ['low blood pressure', 'normal skin thickness', 'normal insulin'],\n",
       " ['low blood pressure', 'normal skin thickness', 'high insulin'],\n",
       " ['low blood pressure', 'high skin thickness', 'low insulin'],\n",
       " ['low blood pressure', 'high skin thickness', 'normal insulin'],\n",
       " ['low blood pressure', 'high skin thickness', 'high insulin'],\n",
       " ['normal blood pressure', 'low skin thickness', 'low insulin'],\n",
       " ['normal blood pressure', 'low skin thickness', 'normal insulin'],\n",
       " ['normal blood pressure', 'low skin thickness', 'high insulin'],\n",
       " ['normal blood pressure', 'normal skin thickness', 'low insulin'],\n",
       " ['normal blood pressure', 'normal skin thickness', 'normal insulin'],\n",
       " ['normal blood pressure', 'normal skin thickness', 'high insulin'],\n",
       " ['normal blood pressure', 'high skin thickness', 'low insulin'],\n",
       " ['normal blood pressure', 'high skin thickness', 'normal insulin'],\n",
       " ['normal blood pressure', 'high skin thickness', 'high insulin'],\n",
       " ['high blood pressure', 'low skin thickness', 'low insulin'],\n",
       " ['high blood pressure', 'low skin thickness', 'normal insulin'],\n",
       " ['high blood pressure', 'low skin thickness', 'high insulin'],\n",
       " ['high blood pressure', 'normal skin thickness', 'low insulin'],\n",
       " ['high blood pressure', 'normal skin thickness', 'normal insulin'],\n",
       " ['high blood pressure', 'normal skin thickness', 'high insulin'],\n",
       " ['high blood pressure', 'high skin thickness', 'low insulin'],\n",
       " ['high blood pressure', 'high skin thickness', 'normal insulin'],\n",
       " ['high blood pressure', 'high skin thickness', 'high insulin'],\n",
       " ['low blood pressure', 'low skin thickness', 'low bmi'],\n",
       " ...]"
      ]
     },
     "execution_count": 27,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "generate_combinations(fields_variations, 3) #example usage"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#creates mimic sample with a specified SampleNumber of -1\n",
    "driver.execute_query( \n",
    "    \"\"\"\n",
    "    Create(mimic:Sample {SampleNumber: -1})\n",
    "    \"\"\"\n",
    "\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#get node id of mimic sample\n",
    "nodeId = driver.execute_query(\n",
    "    \"\"\"\n",
    "    Match (m:Sample) where m.SampleNumber=-1\n",
    "    return id(m)\n",
    "    \"\"\"\n",
    ")[0][0][0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [],
   "source": [
    "#removes all relationships for mimic\n",
    "def clearrels():\n",
    "    driver.execute_query(\n",
    "        \"\"\"\n",
    "        match (n:Sample)-[r]->(m) where n.SampleNumber=-1 delete r\n",
    "        \"\"\"\n",
    "\n",
    "    )"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {},
   "outputs": [],
   "source": [
    "#creates specified relationship between mimic and other node\n",
    "def createrels(relationship, node):\n",
    "    query = f\"\"\"\n",
    "    MATCH (n:Sample {{SampleNumber: -1}}), (m:medical_concept {{name: $nodename}})\n",
    "    MERGE (n)-[r:{relationship}]->(m)\n",
    "    \"\"\"\n",
    "    driver.execute_query(\n",
    "        query,\n",
    "        nodename=node\n",
    "    )\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [],
   "source": [
    "#projects graph with passed in graphName\n",
    "def createGraph(name):\n",
    "    driver.execute_query(\n",
    "        \"\"\"\n",
    "        CALL gds.graph.project(\n",
    "            $graphName,\n",
    "            {\n",
    "                Sample: '*',\n",
    "                medical_concept: '*',\n",
    "                definition: '*',\n",
    "                synonyms: '*'\n",
    "            },\n",
    "            {\n",
    "                HAS_PREGNANCIES: {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                }, \n",
    "                HAS_GLUCOSE: {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                },\n",
    "                HAS_BLOOD_PRESSURE: {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                },\n",
    "                HAS_SKIN_THICKNESS: {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                },\n",
    "                HAS_INSULIN: {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                }, \n",
    "                HAS_BMI: {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                },\n",
    "                HAS_DIABETES_PEDIGREE_FUNCTION: {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                },\n",
    "                HAS_AGE: {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                }, \n",
    "                Definition : {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                },\n",
    "                Synonym: {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                },\n",
    "                embedding_match_node: {\n",
    "                orientation: 'UNDIRECTED'  \n",
    "                }\n",
    "            })\n",
    "        \"\"\",\n",
    "        graphName=name\n",
    "    )\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {},
   "outputs": [],
   "source": [
    "#gets fastrp embedding for mimic\n",
    "def getembedding(gname):\n",
    "    results = driver.execute_query(\n",
    "        \"\"\"\n",
    "        CALL gds.fastRP.stream($graphName, {\n",
    "            embeddingDimension: $embeddingDimension,\n",
    "            randomSeed:42\n",
    "        })\n",
    "        YIELD nodeId, embedding\n",
    "        WHERE nodeId = $nodeId\n",
    "        return nodeId, embedding;\n",
    "        \"\"\",\n",
    "        embeddingDimension=128, graphName=gname, nodeId=nodeId\n",
    "    )\n",
    "\n",
    "    return results[0][0][1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [],
   "source": [
    "#loops through combination list and gets all embeddings for the mimic in a dictionary \n",
    "def getembeddingformimic(clist):\n",
    "    retdict = {} #empty dictionary to hold all resulting embeddings\n",
    "    for i,c in enumerate(clist):\n",
    "       if i%10==0: #if divisible by 10, we delete all graphs in memory \n",
    "           clearGraphs()\n",
    "       graphname = f\"pima{i%10}\"\n",
    "       clearrels() #clear relationships\n",
    "       for node in c: \n",
    "           relationship = relationships[node]\n",
    "           createrels(relationship, node) #create relationship according to combination\n",
    "       createGraph(graphname) #create graph\n",
    "       embedding = getembedding(graphname)\n",
    "       retdict[str(c)]= embedding #add resulting embedding and combination to dictionary\n",
    "    \n",
    "    return retdict\n",
    "           "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#main - applies conmbinations to graph and writes to file \n",
    "import csv \n",
    "for i in range(8):\n",
    "    combinations = generate_combinations(fields_variations, i+1)\n",
    "    embeddings = getembeddingformimic(combinations)\n",
    "    with open(\"data/finalresults.csv\", 'a', newline='') as file:\n",
    "        writer = csv.writer(file)\n",
    "\n",
    "        for key, value in embeddings.items():\n",
    "            writer.writerow([key, value])\n",
    "    clearGraphs()   "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "clearGraphs()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "clearrels()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Applying Embeeddings to Models"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "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>relationships</th>\n",
       "      <th>embedding</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>['low pregnancies']</td>\n",
       "      <td>[0.3559499979019165, 0.017154032364487648, -0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>['normal pregnancies']</td>\n",
       "      <td>[0.21517100930213928, 0.09962455183267593, 0.1...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>['high pregnancies']</td>\n",
       "      <td>[0.1451621651649475, 0.012034144252538681, -0....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>['low glucose']</td>\n",
       "      <td>[0.07380708307027817, 0.07721707224845886, 0.0...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>['normal glucose']</td>\n",
       "      <td>[0.4024839997291565, -0.0163387693464756, -0.0...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "            relationships                                          embedding\n",
       "0     ['low pregnancies']  [0.3559499979019165, 0.017154032364487648, -0....\n",
       "1  ['normal pregnancies']  [0.21517100930213928, 0.09962455183267593, 0.1...\n",
       "2    ['high pregnancies']  [0.1451621651649475, 0.012034144252538681, -0....\n",
       "3         ['low glucose']  [0.07380708307027817, 0.07721707224845886, 0.0...\n",
       "4      ['normal glucose']  [0.4024839997291565, -0.0163387693464756, -0.0..."
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#load combinations and their embeddings \n",
    "embdf = pd.read_csv(\"data/finalresults.csv\")\n",
    "embdf.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "import ast\n",
    "import joblib\n",
    "embdf['embedding'] = embdf['embedding'].apply(ast.literal_eval) #convert embeddings to lists (instead of strings)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "input = np.array(embdf['embedding'].tolist()) #aggregate embedding column"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Random Forest"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "#load best rf model\n",
    "with open('final_models/rfemb.pkl', 'rb') as file:\n",
    "    model = joblib.load(file)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#predict\n",
    "pred = model.predict(input)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "#save prediction to embdf\n",
    "embdf['score_rf'] = pred"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### XGBoost"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [],
   "source": [
    "#load best xgb model\n",
    "with open('final_models/xgbemb.pkl', 'rb') as file:\n",
    "    model = joblib.load(file)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {},
   "outputs": [],
   "source": [
    "#predict\n",
    "pred = model.predict(input)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {},
   "outputs": [],
   "source": [
    "#save prediction to embdf\n",
    "embdf['score_xgb'] = pred"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### SVM"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [],
   "source": [
    "#load best svm model\n",
    "with open('final_models/svmemb.pkl', 'rb') as file:\n",
    "    model = joblib.load(file)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#predict\n",
    "pred = model.predict(input)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [],
   "source": [
    "#save prediction to embdf\n",
    "embdf['score_svm'] = pred"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Naive Bayes"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [],
   "source": [
    "#load best nb model\n",
    "with open('final_models/nbemb.pkl', 'rb') as file:\n",
    "    model = joblib.load(file)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "#predict\n",
    "pred = model.predict(input)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 27,
   "metadata": {},
   "outputs": [],
   "source": [
    "#save prediction to embdf\n",
    "embdf['score_nb'] = pred"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Neural Networks"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {},
   "outputs": [],
   "source": [
    "from tensorflow.keras.models import load_model\n",
    "\n",
    "#load best nn model\n",
    "model = load_model('final_models/nnemb.keras')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\u001b[1m2048/2048\u001b[0m \u001b[32m━━━━━━━━━━━━━━━━━━━━\u001b[0m\u001b[37m\u001b[0m \u001b[1m2s\u001b[0m 772us/step\n"
     ]
    }
   ],
   "source": [
    "#predict\n",
    "pred = model.predict(input)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [],
   "source": [
    "#save prediction to embdf\n",
    "embdf['score_nn'] = pred"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "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>relationships</th>\n",
       "      <th>embedding</th>\n",
       "      <th>score_rf</th>\n",
       "      <th>score_xgb</th>\n",
       "      <th>score_svm</th>\n",
       "      <th>score_nb</th>\n",
       "      <th>score_nn</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>['low pregnancies']</td>\n",
       "      <td>[0.3559499979019165, 0.017154032364487648, -0....</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.170097</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>['normal pregnancies']</td>\n",
       "      <td>[0.21517100930213928, 0.09962455183267593, 0.1...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.298908</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>['high pregnancies']</td>\n",
       "      <td>[0.1451621651649475, 0.012034144252538681, -0....</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.517344</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>['low glucose']</td>\n",
       "      <td>[0.07380708307027817, 0.07721707224845886, 0.0...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.121242</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>['normal glucose']</td>\n",
       "      <td>[0.4024839997291565, -0.0163387693464756, -0.0...</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.048572</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "            relationships                                          embedding  \\\n",
       "0     ['low pregnancies']  [0.3559499979019165, 0.017154032364487648, -0....   \n",
       "1  ['normal pregnancies']  [0.21517100930213928, 0.09962455183267593, 0.1...   \n",
       "2    ['high pregnancies']  [0.1451621651649475, 0.012034144252538681, -0....   \n",
       "3         ['low glucose']  [0.07380708307027817, 0.07721707224845886, 0.0...   \n",
       "4      ['normal glucose']  [0.4024839997291565, -0.0163387693464756, -0.0...   \n",
       "\n",
       "   score_rf  score_xgb  score_svm  score_nb  score_nn  \n",
       "0         0          0          0         0  0.170097  \n",
       "1         0          0          0         0  0.298908  \n",
       "2         0          0          0         0  0.517344  \n",
       "3         0          0          0         0  0.121242  \n",
       "4         0          0          0         0  0.048572  "
      ]
     },
     "execution_count": 31,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#view embdf\n",
    "embdf.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [],
   "source": [
    "#create binary column for nn\n",
    "embdf['score_nn_binary'] = embdf['score_nn'].apply(lambda x: 1 if x>=.5 else 0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {},
   "outputs": [],
   "source": [
    "#save embdf\n",
    "embdf.to_csv(\"data/predictedscores.csv\")"
   ]
  }
 ],
 "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.12.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}