Notice: This page requires JavaScript to function properly.
Please enable JavaScript in your browser settings or update your browser.
Aprende Desafío: Implementación de un Árbol de Decisión | Árbol de Decisión
Clasificación con Python
Sección 3. Capítulo 4
single

single

Desafío: Implementación de un Árbol de Decisión

Desliza para mostrar el menú

En este desafío, utilizarás el conjunto de datos Titanic, que contiene información sobre los pasajeros del Titanic, incluyendo su edad, sexo, tamaño de la familia y más. El objetivo es predecir si un pasajero sobrevivió o no.

1234
import pandas as pd df = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/b71ff7ac-3932-41d2-a4d8-060e24b00129/titanic.csv') print(df.head())

Para implementar el Árbol de Decisión, puedes utilizar DecisionTreeClassifier de sklearn:

DecisionTreeClass

Construcción de un árbol de decisión y búsqueda de los mejores valores para max_depth y min_samples_leaf mediante búsqueda en cuadrícula.

Tarea

Desliza para comenzar a programar

Se proporciona un conjunto de datos del Titanic almacenado como un DataFrame en la variable df.

  • Inicializar un modelo de Árbol de Decisión y almacenarlo en la variable decision_tree.
  • Crear un diccionario para que GridSearchCV itere por los valores [1, 2, 3, 4, 5, 6, 7] para max_depth y [1, 2, 4, 6] para min_samples_leaf, y almacenarlo en la variable param_grid.
  • Inicializar y entrenar un objeto GridSearchCV, establecer el número de particiones en 10 y almacenar el modelo entrenado en la variable grid_cv.

Solución

Switch to desktopCambia al escritorio para practicar en el mundo realContinúe desde donde se encuentra utilizando una de las siguientes opciones
¿Todo estuvo claro?

¿Cómo podemos mejorarlo?

¡Gracias por tus comentarios!

Sección 3. Capítulo 4
single

single

Pregunte a AI

expand

Pregunte a AI

ChatGPT

Pregunte lo que quiera o pruebe una de las preguntas sugeridas para comenzar nuestra charla

some-alt