Entradas

Mostrando las entradas etiquetadas como Titanic

Cómo crear un árbol de decisión en R desde cero.

Imagen
En este artículo, vamos a crear un árbol de decisión para predecir la supervivencia de los pasajeros en función de tres variables: age, sex y Pclass (passenger class). Las variables usadas en el árbol son relevantes, si bien no son el modelo perfecto. Ante todo, queremos un modelo simple que nos aporte compresión sobre el problema y nos permita crear modelos más avanzados. CARGA DE LIBRERÍAS y DATOS Cargamos las librerias y los datos de entrenamiento library(tidyverse) library(caTools) train <- read.csv("train.csv") test <- read.csv("test.csv") head(train) ## PassengerId Survived Pclass ## 1 1 0 3 ## 2 2 1 1 ## 3 3 1 3 ## 4 4 1 1 ## 5 5 0 3 ## 6 6 0 3 ## Name Sex Age SibSp Parch ## 1 Braund, Mr. Owen Harris 1 22 1 0 ## 2 Cumi...