En este capítulo, vas a usar una versión del conjunto de datos Wisconsin Breast Cancer. Este conjunto plantea un problema clásico de clasificación binaria: el 50% de las muestras son benignas, el 50% malignas, y el reto es identificar cuáles son cuáles.

Este conjunto es interesante porque muchos predictores contienen valores ausentes y la mayoría de las filas tiene al menos un valor faltante. Esto complica el modelado, porque la mayoría de los algoritmos de Machine Learning no gestionan valores ausentes de forma nativa. Por ejemplo, tu primera idea podría ser ajustar una regresión logística, pero antes necesitas una estrategia para tratar los NA.

Por suerte, la función train() de caret incluye el argumento preProcess, que te permite indicar que se use imputación por la mediana para rellenar los valores ausentes. En capítulos anteriores, creaste modelos con train() usando fórmulas como y ~ .. Una alternativa es especificar los argumentos x y y de train(), donde x es un objeto con muestras en filas y características en columnas y y es un vector numérico o factor que contiene los resultados. Dicho de otro modo, x es una matriz o data frame que contiene todo el conjunto de datos que usarías en el argumento data de una llamada a lm(), por ejemplo, pero excluye la columna de la variable respuesta; y es un vector que contiene solo la columna de la variable respuesta.

Para este ejercicio, el argumento x de train() está cargado en tu espacio de trabajo como breast_cancer_x y y como breast_cancer_y.

Imputación por mediana vs. omitir filas

Machine Learning con caret en R

Ejercicio interactivo práctico