1 Objetivo

Implementar el modelo de árbol de clasificación con datos relacionados a una condición de salud de las personas para predecir anomalías de corazón y evaluar la exactitud del modelo mediante la matriz de confusión.

2 Descripción

Se cargan librerías y se descargan los datos: https://raw.githubusercontent.com/rpizarrog/Analisis-Inteligente-de-datos/main/datos/heart_2020_cleaned.csv

Los datos están relacionados con aspectos médicos y son valores numéricos de varias variables que caracterizan el estado de salud de 319,795 personas.

Se construye un modelo supervisado basado en el algoritmo de árbol de clasificación para resolver la tarea de clasificación binaria e identificar si una persona padece del corazón o no.

Se construyen datos de entrenamiento y validación al 80% y 20% cada uno.

Se desarrollan los modelos de:

  • Regresión Logística binaria

  • Árbol de Clasificación tipo class

  • K Means

  • SVM Lineal

  • SVM Polinomial

  • SVM Radial

Los modelo se aceptan si tienen un valor de exactitud por encima del 70%..

3 Fundamento teórico

4 Desarrollo

4.1 Cargar librerías

library(readr)
library(dplyr)
library(caret)
library(rpart)
library(rpart.plot)
library(knitr)
library(e1071)        # Vectores de Soporte SVM
library(rpart)        # Arboles de clasificación

4.2 Cargar datos

Cargar datos de manera local.

# datos <- read.csv("https://raw.githubusercontent.com/rpizarrog/Machine-Learning-con-R/main/datos/heart_2020_cleaned.csv")
datos <- read.csv("https://raw.githubusercontent.com/rpizarrog/Analisis-Inteligente-de-datos/main/datos/heart_2020_cleaned.csv", encoding = "UTF-8", stringsAsFactors = TRUE)

4.3 Explorar datos

str(datos)
## 'data.frame':    319795 obs. of  18 variables:
##  $ HeartDisease    : Factor w/ 2 levels "No","Yes": 1 1 1 1 1 2 1 1 1 1 ...
##  $ BMI             : num  16.6 20.3 26.6 24.2 23.7 ...
##  $ Smoking         : Factor w/ 2 levels "No","Yes": 2 1 2 1 1 2 1 2 1 1 ...
##  $ AlcoholDrinking : Factor w/ 2 levels "No","Yes": 1 1 1 1 1 1 1 1 1 1 ...
##  $ Stroke          : Factor w/ 2 levels "No","Yes": 1 2 1 1 1 1 1 1 1 1 ...
##  $ PhysicalHealth  : num  3 0 20 0 28 6 15 5 0 0 ...
##  $ MentalHealth    : num  30 0 30 0 0 0 0 0 0 0 ...
##  $ DiffWalking     : Factor w/ 2 levels "No","Yes": 1 1 1 1 2 2 1 2 1 2 ...
##  $ Sex             : Factor w/ 2 levels "Female","Male": 1 1 2 1 1 1 1 1 1 2 ...
##  $ AgeCategory     : Factor w/ 13 levels "18-24","25-29",..: 8 13 10 12 5 12 11 13 13 10 ...
##  $ Race            : Factor w/ 6 levels "American Indian/Alaskan Native",..: 6 6 6 6 6 3 6 6 6 6 ...
##  $ Diabetic        : Factor w/ 4 levels "No","No, borderline diabetes",..: 3 1 3 1 1 1 1 3 2 1 ...
##  $ PhysicalActivity: Factor w/ 2 levels "No","Yes": 2 2 2 1 2 1 2 1 1 2 ...
##  $ GenHealth       : Factor w/ 5 levels "Excellent","Fair",..: 5 5 2 3 5 2 2 3 2 3 ...
##  $ SleepTime       : num  5 7 8 6 8 12 4 9 5 10 ...
##  $ Asthma          : Factor w/ 2 levels "No","Yes": 2 1 2 1 1 1 2 2 1 1 ...
##  $ KidneyDisease   : Factor w/ 2 levels "No","Yes": 1 1 1 1 1 1 1 1 2 1 ...
##  $ SkinCancer      : Factor w/ 2 levels "No","Yes": 2 1 1 2 1 1 2 1 1 1 ...
summary(datos)
##  HeartDisease      BMI        Smoking      AlcoholDrinking Stroke      
##  No :292422   Min.   :12.02   No :187887   No :298018      No :307726  
##  Yes: 27373   1st Qu.:24.03   Yes:131908   Yes: 21777      Yes: 12069  
##               Median :27.34                                            
##               Mean   :28.33                                            
##               3rd Qu.:31.42                                            
##               Max.   :94.85                                            
##                                                                        
##  PhysicalHealth    MentalHealth    DiffWalking      Sex        
##  Min.   : 0.000   Min.   : 0.000   No :275385   Female:167805  
##  1st Qu.: 0.000   1st Qu.: 0.000   Yes: 44410   Male  :151990  
##  Median : 0.000   Median : 0.000                               
##  Mean   : 3.372   Mean   : 3.898                               
##  3rd Qu.: 2.000   3rd Qu.: 3.000                               
##  Max.   :30.000   Max.   :30.000                               
##                                                                
##       AgeCategory                                 Race       
##  65-69      : 34151   American Indian/Alaskan Native:  5202  
##  60-64      : 33686   Asian                         :  8068  
##  70-74      : 31065   Black                         : 22939  
##  55-59      : 29757   Hispanic                      : 27446  
##  50-54      : 25382   Other                         : 10928  
##  80 or older: 24153   White                         :245212  
##  (Other)    :141601                                          
##                     Diabetic      PhysicalActivity     GenHealth     
##  No                     :269653   No : 71838       Excellent: 66842  
##  No, borderline diabetes:  6781   Yes:247957       Fair     : 34677  
##  Yes                    : 40802                    Good     : 93129  
##  Yes (during pregnancy) :  2559                    Poor     : 11289  
##                                                    Very good:113858  
##                                                                      
##                                                                      
##    SleepTime      Asthma       KidneyDisease SkinCancer  
##  Min.   : 1.000   No :276923   No :308016    No :289976  
##  1st Qu.: 6.000   Yes: 42872   Yes: 11779    Yes: 29819  
##  Median : 7.000                                          
##  Mean   : 7.097                                          
##  3rd Qu.: 8.000                                          
##  Max.   :24.000                                          
## 

4.4 Limpiar datos

No es necesario alguna transformación

4.5 Las variables de interés

Todas las variables son de entrada o variables independientes:

  • BMI”: Indice de masa corporal con valores entre 12.02 y 94.85.

  • Smoking”: Si la persona es fumadora o no con valores categóritos de ‘Yes’ o ‘No’.

  • AlcoholDrinking” : Si consume alcohol o no, con valores categóricos de ‘Yes’ o ‘No’.

  • Stroke”: Si padece alguna anomalía cerebrovascular, apoplejia o algo similar, con valores categóricos de ‘Yes’ o ‘No’.

  • PhysicalHealth” Estado físico en lo general con valores entre 0 y 30.

  • MentalHealth”. Estado mental en lo general con valores entre 0 y 30.

  • DiffWalking” . Que si se le dificulta caminar o tiene algún padecimiento al caminar, con valores categóritoc de ‘Yes’ o ‘No’.

  • Sex”: Género de la persona, con valores de ‘Female’ y ‘Male’ para distinguir al género femenino y masculino respectivamente.

  • AgeCategory”: Una clasificación de la edad de la persona de entre 18 y 80 años. La primera categoría con un rango de edad entre 18-24, a partir de 25 con rangos de 5 en 5 hasta la clase de 75-80 y una última categoría mayores de 80 años.

  • Race”. Raza u origen de la persona con valores categóricos de ‘American Indian/Alaskan Native’, ’Asian’,’Black’, ’Hispanic’, ’Other’ y’White’.

  • Diabetic”. Si padece o ha padecido de diabetes en cuatro condiciones siendo Yes y No para si o no: ‘No’, ‘borderline diabetes’ condición antes de detectarse diabetes tipo 2, ‘Yes’, y ‘Yes (during pregnancy)’ durante embarazo.

  • PhysicalActivity” que si realiza actividad física, con valores categóricos de ‘Yes’ o ‘No’.

  • GenHealth”: EStado general de salud de la persona con valores categóricos de ‘Excellent’, ‘Very good’, ’Good’, ’Fair’ y ’Poor’ con significado en español de excelente, muy buena, buena, regular y pobre o deficiente.

  • SleepTime”: valor numérico de las horas de sueño u horas que duerme la persona con valores en un rango entre 1 y 24.

  • Asthma”: si padece de asma o no, con valores categóricos de ‘Yes’ o ‘No’.

  • KidneyDisease”: si tiene algún padecimiento en los riñones, con valores categóricos de ‘Yes’ o ‘No’.

  • SkinCancer”: si padece algún tipo de cáncer de piel, con valores categóricos de ‘Yes’ o ‘No’.

La variable de interés como dependiente o variable de salida es la de daño al corazón (HeartDisease), con valores categóricos de ‘Yes’ o ‘No’.

4.6 Datos de entrenamiento y validación

Se parten los datos en en datos de entrenamiento con el 80% y datos de validación con el 20%.

set.seed(1307)
entrena <- createDataPartition(y = datos$HeartDisease, 
                               p = 0.8, 
                               list = FALSE, 
                               times = 1)
# Datos entrenamiento
datos.entrenamiento <- datos[entrena, ]  # [renglones, columna]
# Datos validación
datos.validacion <- datos[-entrena, ]

4.6.1 Datos de entrenamiento

Se muestran los primeros 20 registros datos de entrenamiento

kable(head(datos.entrenamiento, 20), caption = "Primeros 20 registros de datos de entrenamiento")
Primeros 20 registros de datos de entrenamiento
HeartDisease BMI Smoking AlcoholDrinking Stroke PhysicalHealth MentalHealth DiffWalking Sex AgeCategory Race Diabetic PhysicalActivity GenHealth SleepTime Asthma KidneyDisease SkinCancer
1 No 16.60 Yes No No 3 30 No Female 55-59 White Yes Yes Very good 5 Yes No Yes
2 No 20.34 No No Yes 0 0 No Female 80 or older White No Yes Very good 7 No No No
4 No 24.21 No No No 0 0 No Female 75-79 White No No Good 6 No No Yes
5 No 23.71 No No No 28 0 Yes Female 40-44 White No Yes Very good 8 No No No
6 Yes 28.87 Yes No No 6 0 Yes Female 75-79 Black No No Fair 12 No No No
7 No 21.63 No No No 15 0 No Female 70-74 White No Yes Fair 4 Yes No Yes
8 No 31.64 Yes No No 5 0 Yes Female 80 or older White Yes No Good 9 Yes No No
10 No 40.69 No No No 0 0 Yes Male 65-69 White No Yes Good 10 No No No
11 Yes 34.30 Yes No No 30 0 Yes Male 60-64 White Yes No Poor 15 Yes No No
12 No 28.71 Yes No No 0 0 No Female 55-59 White No Yes Very good 5 No No No
13 No 28.37 Yes No No 0 0 Yes Male 75-79 White Yes Yes Very good 8 No No No
14 No 28.15 No No No 7 0 Yes Female 80 or older White No No Good 7 No No No
15 No 29.29 Yes No No 0 30 Yes Female 60-64 White No No Good 5 No No No
16 No 29.18 No No No 1 0 No Female 50-54 White No Yes Very good 6 No No No
17 No 26.26 No No No 5 2 No Female 70-74 White No No Very good 10 No No No
18 No 22.59 Yes No No 0 30 Yes Male 70-74 White No, borderline diabetes Yes Good 8 No No No
19 No 29.86 Yes No No 0 0 Yes Female 75-79 Black Yes No Fair 5 No Yes No
20 No 18.13 No No No 0 0 No Male 80 or older White No Yes Excellent 8 No No Yes
21 No 21.16 No No No 0 0 No Female 80 or older Black No, borderline diabetes No Good 8 No No No
23 No 26.17 Yes No No 0 15 No Female 45-49 White No Yes Very good 6 No No No

4.6.2 Datos de validación

Se muestran los primeros 20 registros de datos de validación .

kable(head(datos.validacion, 20), caption = "Primeros 20 registros de datos de validación")
Primeros 20 registros de datos de validación
HeartDisease BMI Smoking AlcoholDrinking Stroke PhysicalHealth MentalHealth DiffWalking Sex AgeCategory Race Diabetic PhysicalActivity GenHealth SleepTime Asthma KidneyDisease SkinCancer
3 No 26.58 Yes No No 20 30 No Male 65-69 White Yes Yes Fair 8 Yes No No
9 No 26.45 No No No 0 0 No Female 80 or older White No, borderline diabetes No Fair 5 No Yes No
22 No 28.90 No No No 2 5 No Female 70-74 White Yes No Very good 7 No No No
24 No 25.82 Yes No No 0 30 No Male 80 or older White Yes Yes Fair 8 No No No
29 No 24.89 No No No 1 0 No Female 55-59 White No Yes Very good 7 No No No
32 No 30.67 No No No 4 4 Yes Female 80 or older White No Yes Fair 8 Yes No No
37 No 24.62 No No No 5 0 No Female 80 or older White No Yes Good 6 No No No
38 No 20.43 Yes No No 3 3 No Female 65-69 White No Yes Fair 5 Yes No Yes
39 No 27.37 Yes No No 0 0 No Female 80 or older White No No Very good 8 No No No
45 No 30.34 Yes No No 30 10 Yes Male 55-59 White No, borderline diabetes No Fair 6 No No No
54 No 32.61 Yes No No 0 0 No Female 65-69 White No Yes Very good 7 No No No
57 No 32.55 Yes No No 0 0 No Male 75-79 White No No Very good 8 No No No
59 No 29.55 No No No 0 0 No Male 70-74 White Yes Yes Good 7 No No No
62 No 21.03 No No No 1 0 No Female 80 or older White No Yes Excellent 8 No No No
68 No 22.04 No No No 0 0 No Male 65-69 White No Yes Very good 8 No No No
73 No 33.75 No No No 0 5 No Male 50-54 White Yes Yes Good 10 Yes No No
78 Yes 34.33 No No No 0 0 No Female 55-59 White No Yes Good 8 No No No
84 No 28.25 No No No 0 3 Yes Male 75-79 White No Yes Good 8 No No No
86 No 23.78 No No No 0 5 No Female 70-74 White No Yes Excellent 8 No No No
89 No 32.81 Yes No No 0 0 Yes Female 70-74 White Yes Yes Good 5 No No No

4.7 Arbol de Clasificación

Se construye el modelo con los datos de entrenamiento mediante la función rpart().

El árbol no se puede visualizar cn todos los registros de los datos de entrenamiento, se hicieron las pruebas y se logra visualizar aproximadamente con 2000 registros de una muestra a partir de los datos de entrenamiento.

Si se construye el modelo con todos los registros de los datos de entrenamiento, pero no se puede observar la visualización del árbol y sus ramificaciones, razón por la cual se hace con una muestra de los datos de entrenamiento.

muestra <- sample(x = 1:nrow(datos.entrenamiento), size = 2000, replace = FALSE)
modelo.ac = rpart(data = datos.entrenamiento[muestra,],formula =    HeartDisease ~ .)

4.7.1 Resumen y/o estadísticos del modelo

El resumen del modelo muestra algunos estadísticos importantes:

summary(modelo.ac)
## Call:
## rpart(formula = HeartDisease ~ ., data = datos.entrenamiento[muestra, 
##     ])
##   n= 2000 
## 
##           CP nsplit rel error   xerror       xstd
## 1 0.01013514      0 1.0000000 1.000000 0.07909966
## 2 0.01000000      8 0.9121622 1.094595 0.08244295
## 
## Variable importance
##      GenHealth    AgeCategory PhysicalHealth            BMI       Diabetic 
##             39             22             14             11              6 
##   MentalHealth    DiffWalking           Race            Sex  KidneyDisease 
##              3              1              1              1              1 
## 
## Node number 1: 2000 observations,    complexity param=0.01013514
##   predicted class=No   expected loss=0.074  P(node) =1
##     class counts:  1852   148
##    probabilities: 0.926 0.074 
##   left son=2 (1725 obs) right son=3 (275 obs)
##   Primary splits:
##       GenHealth      splits as  LRLRL, improve=19.14538, (0 missing)
##       DiffWalking    splits as  LR, improve=14.24656, (0 missing)
##       PhysicalHealth < 8.5    to the left,  improve=13.34158, (0 missing)
##       Diabetic       splits as  LLRL, improve=12.13351, (0 missing)
##       AgeCategory    splits as  LLLLLLLLRRRRR, improve=11.80676, (0 missing)
##   Surrogate splits:
##       PhysicalHealth < 17.5   to the left,  agree=0.890, adj=0.204, (0 split)
##       BMI            < 16.105 to the right, agree=0.864, adj=0.007, (0 split)
##       SleepTime      < 13     to the left,  agree=0.864, adj=0.007, (0 split)
## 
## Node number 2: 1725 observations
##   predicted class=No   expected loss=0.04637681  P(node) =0.8625
##     class counts:  1645    80
##    probabilities: 0.954 0.046 
## 
## Node number 3: 275 observations,    complexity param=0.01013514
##   predicted class=No   expected loss=0.2472727  P(node) =0.1375
##     class counts:   207    68
##    probabilities: 0.753 0.247 
##   left son=6 (94 obs) right son=7 (181 obs)
##   Primary splits:
##       AgeCategory    splits as  LLLLLLRLRRRRR, improve=8.529484, (0 missing)
##       GenHealth      splits as  -L-R-, improve=4.143086, (0 missing)
##       Diabetic       splits as  LLRL, improve=3.681166, (0 missing)
##       Sex            splits as  LR, improve=2.733953, (0 missing)
##       PhysicalHealth < 6.5    to the left,  improve=2.572245, (0 missing)
##   Surrogate splits:
##       BMI          < 37.32  to the right, agree=0.687, adj=0.085, (0 split)
##       Race         splits as  LRRLLR,     agree=0.676, adj=0.053, (0 split)
##       MentalHealth < 15.5   to the right, agree=0.673, adj=0.043, (0 split)
##       Diabetic     splits as  RRRL,       agree=0.669, adj=0.032, (0 split)
## 
## Node number 6: 94 observations
##   predicted class=No   expected loss=0.07446809  P(node) =0.047
##     class counts:    87     7
##    probabilities: 0.926 0.074 
## 
## Node number 7: 181 observations,    complexity param=0.01013514
##   predicted class=No   expected loss=0.3370166  P(node) =0.0905
##     class counts:   120    61
##    probabilities: 0.663 0.337 
##   left son=14 (55 obs) right son=15 (126 obs)
##   Primary splits:
##       BMI            < 25.615 to the left,  improve=2.966518, (0 missing)
##       Sex            splits as  LR, improve=2.929183, (0 missing)
##       AgeCategory    splits as  ------L-LLRRL, improve=2.177522, (0 missing)
##       SleepTime      < 4.5    to the left,  improve=1.418737, (0 missing)
##       PhysicalHealth < 6.5    to the left,  improve=1.252412, (0 missing)
##   Surrogate splits:
##       Race splits as  RLRRRR, agree=0.707, adj=0.036, (0 split)
## 
## Node number 14: 55 observations
##   predicted class=No   expected loss=0.2  P(node) =0.0275
##     class counts:    44    11
##    probabilities: 0.800 0.200 
## 
## Node number 15: 126 observations,    complexity param=0.01013514
##   predicted class=No   expected loss=0.3968254  P(node) =0.063
##     class counts:    76    50
##    probabilities: 0.603 0.397 
##   left son=30 (79 obs) right son=31 (47 obs)
##   Primary splits:
##       AgeCategory    splits as  ------L-LLRRL, improve=2.735990, (0 missing)
##       BMI            < 26.34  to the right, improve=1.411477, (0 missing)
##       SleepTime      < 4.5    to the left,  improve=1.262376, (0 missing)
##       PhysicalHealth < 6.5    to the left,  improve=1.241316, (0 missing)
##       Diabetic       splits as  LLR-, improve=1.146934, (0 missing)
##   Surrogate splits:
##       BMI      < 25.76  to the right, agree=0.635, adj=0.021, (0 split)
##       Race     splits as  LLLLRL,     agree=0.635, adj=0.021, (0 split)
##       Diabetic splits as  LRL-,       agree=0.635, adj=0.021, (0 split)
## 
## Node number 30: 79 observations,    complexity param=0.01013514
##   predicted class=No   expected loss=0.3164557  P(node) =0.0395
##     class counts:    54    25
##    probabilities: 0.684 0.316 
##   left son=60 (31 obs) right son=61 (48 obs)
##   Primary splits:
##       PhysicalHealth < 6.5    to the left,  improve=1.5414630, (0 missing)
##       MentalHealth   < 2.5    to the left,  improve=1.3554200, (0 missing)
##       Smoking        splits as  RL,         improve=0.8772152, (0 missing)
##       Sex            splits as  LR,         improve=0.8772152, (0 missing)
##       BMI            < 39.015 to the left,  improve=0.7714181, (0 missing)
##   Surrogate splits:
##       MentalHealth < 1      to the left,  agree=0.734, adj=0.323, (0 split)
##       DiffWalking  splits as  LR,         agree=0.684, adj=0.194, (0 split)
##       BMI          < 26.62  to the left,  agree=0.633, adj=0.065, (0 split)
##       SleepTime    < 9.5    to the right, agree=0.620, adj=0.032, (0 split)
## 
## Node number 31: 47 observations,    complexity param=0.01013514
##   predicted class=Yes  expected loss=0.4680851  P(node) =0.0235
##     class counts:    22    25
##    probabilities: 0.468 0.532 
##   left son=62 (39 obs) right son=63 (8 obs)
##   Primary splits:
##       BMI              < 27.33  to the right, improve=0.9170758, (0 missing)
##       Smoking          splits as  LR,         improve=0.8499075, (0 missing)
##       SleepTime        < 6.5    to the left,  improve=0.7798209, (0 missing)
##       PhysicalHealth   < 7      to the left,  improve=0.6335786, (0 missing)
##       PhysicalActivity splits as  LR,         improve=0.4951644, (0 missing)
## 
## Node number 60: 31 observations
##   predicted class=No   expected loss=0.1935484  P(node) =0.0155
##     class counts:    25     6
##    probabilities: 0.806 0.194 
## 
## Node number 61: 48 observations,    complexity param=0.01013514
##   predicted class=No   expected loss=0.3958333  P(node) =0.024
##     class counts:    29    19
##    probabilities: 0.604 0.396 
##   left son=122 (30 obs) right son=123 (18 obs)
##   Primary splits:
##       Diabetic       splits as  LLR-,       improve=2.6694440, (0 missing)
##       PhysicalHealth < 17.5   to the right, improve=1.7187730, (0 missing)
##       Race           splits as  RLRLLR,     improve=1.0489260, (0 missing)
##       BMI            < 39.015 to the left,  improve=1.0083330, (0 missing)
##       Sex            splits as  LR,         improve=0.7440476, (0 missing)
##   Surrogate splits:
##       BMI             < 39.015 to the left,  agree=0.667, adj=0.111, (0 split)
##       KidneyDisease   splits as  LR,         agree=0.667, adj=0.111, (0 split)
##       AlcoholDrinking splits as  LR,         agree=0.646, adj=0.056, (0 split)
##       PhysicalHealth  < 14.5   to the right, agree=0.646, adj=0.056, (0 split)
##       MentalHealth    < 1      to the right, agree=0.646, adj=0.056, (0 split)
## 
## Node number 62: 39 observations,    complexity param=0.01013514
##   predicted class=No   expected loss=0.4871795  P(node) =0.0195
##     class counts:    20    19
##    probabilities: 0.513 0.487 
##   left son=124 (18 obs) right son=125 (21 obs)
##   Primary splits:
##       PhysicalHealth < 2.5    to the left,  improve=1.5824180, (0 missing)
##       Smoking        splits as  LR,         improve=1.0268620, (0 missing)
##       MentalHealth   < 12     to the left,  improve=0.7538462, (0 missing)
##       SleepTime      < 6.5    to the left,  improve=0.4676990, (0 missing)
##       GenHealth      splits as  -L-R-,      improve=0.3423519, (0 missing)
##   Surrogate splits:
##       GenHealth    splits as  -L-R-,      agree=0.718, adj=0.389, (0 split)
##       Sex          splits as  RL,         agree=0.692, adj=0.333, (0 split)
##       DiffWalking  splits as  LR,         agree=0.667, adj=0.278, (0 split)
##       BMI          < 31.05  to the left,  agree=0.641, adj=0.222, (0 split)
##       MentalHealth < 1.5    to the left,  agree=0.615, adj=0.167, (0 split)
## 
## Node number 63: 8 observations
##   predicted class=Yes  expected loss=0.25  P(node) =0.004
##     class counts:     2     6
##    probabilities: 0.250 0.750 
## 
## Node number 122: 30 observations
##   predicted class=No   expected loss=0.2666667  P(node) =0.015
##     class counts:    22     8
##    probabilities: 0.733 0.267 
## 
## Node number 123: 18 observations
##   predicted class=Yes  expected loss=0.3888889  P(node) =0.009
##     class counts:     7    11
##    probabilities: 0.389 0.611 
## 
## Node number 124: 18 observations
##   predicted class=No   expected loss=0.3333333  P(node) =0.009
##     class counts:    12     6
##    probabilities: 0.667 0.333 
## 
## Node number 125: 21 observations
##   predicted class=Yes  expected loss=0.3809524  P(node) =0.0105
##     class counts:     8    13
##    probabilities: 0.381 0.619

Entonces una posible predicción sería siguiendo las reglas de asociación y condicionales del modelo.

4.7.2 Visualizar árbol de clasificación

prp(modelo.ac, main = "Arbol de Clasificación")

4.7.3 Generar predicciones del modelo regresión logística

Se generan predicciones con datos de validación con el argumento class de clasificación, es decir, Yes o No.

prediciones_ac = predict(object = modelo.ac,newdata = datos.validacion, type = "class")

4.7.4 Predicciones

Head(predicciones, 20) los primeros 20 predicciones

head(prediciones_ac, 20)
##   3   9  22  24  29  32  37  38  39  45  54  57  59  62  68  73  78  84  86  89 
## Yes  No  No  No  No  No  No  No  No  No  No  No  No  No  No  No  No  No  No  No 
## Levels: No Yes

4.7.5 Generar tabla comparativa

Se construye una tabla comparativa con los valores de interés

t_comparativa = data.frame("real" = datos.validacion[,c('HeartDisease')],"prediccion"= prediciones_ac)
# t_comparativa <- t_comparativa %>%
#  mutate(heartDiseasePred = 
top20 = head(t_comparativa,20)
kable(top20,caption = 'Primeros 20 registros')
Primeros 20 registros
real prediccion
3 No Yes
9 No No
22 No No
24 No No
29 No No
32 No No
37 No No
38 No No
39 No No
45 No No
54 No No
57 No No
59 No No
62 No No
68 No No
73 No No
78 Yes No
84 No No
86 No No
89 No No

4.7.6 Evaluando el modelo

Una matriz de confusión es una herramienta que permite evaluación de un modelo de clasificación

Cada columna de la matriz representa el número de predicciones de cada clase, mientras que cada fila representa a las instancias en la clase real.

Uno de los beneficios de las matrices de confusión es que facilitan ver si el sistema está confundiendo las diferentes clases o resultados.

Hay que encontrar a cuantos casos se le atinaron utilizando los datos de validación y con ello encontrar el porcentaje de aciertos.

Se puede evaluar el modelo con la matriz de confusión interpretando algunos estadísticos:

Se evalúa el modelo de acuerdo a estas condiciones:

  • Accuracy o exactitud \[ accuracy = \frac{VP + VN}{VP+FP+FN+VN} \\ n = VP+FP+FN+VN \]

  • Precision o precisión

\[ precision = \frac{VP}{VP + FP} \]

  • Recall o recuperación \[ recall = \frac{VP}{VP + FN} \]

  • Especificity o especificidad (tasa de verdaderos negativos)

\[ especificity = \frac{VN}{VN + FP} \]

4.7.6.1 Construyendo la matriz de confusión del modelo de regresión logística

Factorizar las columnas

Factorizar en R significa categorizar con la función “as.factor” o “factor”

Se muestra a tabla con las columnas de interés para interpretar las predicciones.

t_comparativa$real = as.factor(t_comparativa$real)
t_comparativa$prediccion = as.factor(t_comparativa$prediccion)
kable(head(t_comparativa, 20), caption = "Tabla comparativa, primeros 20 registros")
Tabla comparativa, primeros 20 registros
real prediccion
3 No Yes
9 No No
22 No No
24 No No
29 No No
32 No No
37 No No
38 No No
39 No No
45 No No
54 No No
57 No No
59 No No
62 No No
68 No No
73 No No
78 Yes No
84 No No
86 No No
89 No No

Creando de la matriz de confusión con la función confusionMatrix() de la librería caret con las variables de interés: “real” y “prediccion”, que representan los valores reales y las predicciones respectivamente.

matrixConfusion <- confusionMatrix(t_comparativa$real,t_comparativa$prediccion)
matrixConfusion
## Confusion Matrix and Statistics
## 
##           Reference
## Prediction    No   Yes
##        No  57444  1040
##        Yes  4872   602
##                                           
##                Accuracy : 0.9076          
##                  95% CI : (0.9053, 0.9098)
##     No Information Rate : 0.9743          
##     P-Value [Acc > NIR] : 1               
##                                           
##                   Kappa : 0.135           
##                                           
##  Mcnemar's Test P-Value : <2e-16          
##                                           
##             Sensitivity : 0.9218          
##             Specificity : 0.3666          
##          Pos Pred Value : 0.9822          
##          Neg Pred Value : 0.1100          
##              Prevalence : 0.9743          
##          Detection Rate : 0.8982          
##    Detection Prevalence : 0.9144          
##       Balanced Accuracy : 0.6442          
##                                           
##        'Positive' Class : No              
## 

El valor estadístico de Accuracy = Exactitud igual a 0.9069 significa un valor aproximado del 90.69; se interpreta que de cada 100 el modelo acierta en la predicción el 90.69% de las ocasiones.

El modelo se construyó solo con una muestra de 2000 registros de los datos de entrenamiento.

Si la métrica era que debiera tener un valor por encima del 70% el modelo se acepta pero debe compararse contra otro modelo de clasificación para ver cual es más eficiente en relación tan solo en el estadístico de exactitud.

Este valor de Accuracy = Exactitud deberá compararse contra otros modelos.

4.8 Predicciones con datos nuevos

Se crea un registro de una persona con ciertas condiciones de salud.

BMI <- 38
Smoking <- 'Yes'
AlcoholDrinking = 'Yes'
Stroke <- 'Yes'
PhysicalHealth <- 2
MentalHealth = 5
DiffWalking = 'Yes'
Sex = 'Male'
AgeCategory = '70-74'
Race = 'Black'
Diabetic <- 'Yes'
PhysicalActivity = "No"
GenHealth = "Fair"
SleepTime = 12
Asthma = "Yes"
KidneyDisease = "Yes"
SkinCancer = 'No'
persona <- data.frame(BMI,Smoking, AlcoholDrinking, Stroke, PhysicalHealth, MentalHealth, DiffWalking, Sex, AgeCategory, Race, Diabetic, PhysicalActivity, GenHealth, SleepTime, Asthma, KidneyDisease, SkinCancer)
persona
##   BMI Smoking AlcoholDrinking Stroke PhysicalHealth MentalHealth DiffWalking
## 1  38     Yes             Yes    Yes              2            5         Yes
##    Sex AgeCategory  Race Diabetic PhysicalActivity GenHealth SleepTime Asthma
## 1 Male       70-74 Black      Yes               No      Fair        12    Yes
##   KidneyDisease SkinCancer
## 1           Yes         No

Se hace la predicción con estos valores:

prediccion <- predict(object = modelo.ac, newdata = persona, type = "class")
prediccion
##  1 
## No 
## Levels: No Yes
# prediccion <- prediccion$fit
# prediccion

Entonces la predicción es:

Si la predicción es ‘No’ entonces no tienen afección del corazón, en caso contrario de ‘Yes’ entonces implica que si tiene daño del corazón.

5 Interpretación

Como en el caso anterio, con la semilla 1307 existen mas numeros de falsos Negativos, 4872, que de falsos positivos, 602 La presicion resultante es de 90.76%, lo cual es mayor al requerido de 70, esto al igual que en el anterior caso si bien puede parecer poco al ser algo importante de salud, en realidad es bastante bueno.

Yo los veo como una herramienta de prediccion mas y dependera de los especialistas si se utiliza solo como un como un paso adicional para buscar afecciones al corazon, lo veo como algo util y se puede buscar mejorar el modelo o cambiarlo para obtener algo mayor al 95% que es donde muchas personas empiezan a sentir una seguridad mas grande.

El modelo de arbol de clasificacion me gusta visualmente en r studio pero esto es independiente de los resultados que se pueden obtener.

6 Bibliografía