Del curso: Python para data scientist avanzado

Accede al curso completo hoy mismo

Únete hoy para acceder a más de 24.700 cursos impartidos por expertos del sector.

XGboost en Python

XGboost en Python

Vamos a ver un ejemplo comentado de código de cómo ajustar un árbol de regresión utilizando XGBoost. Lo primero que tienes que hacer, si no lo has hecho ya, es instalar el paquete. Este comando de aquí puedes usarlo en la consola y se va a instalar directamente a Python. Es un paquete que a veces da problemas. Así que si te da algún error, buscas por Internet y encontrarás tutoriales de cómo descargar directamente el paquete y cargarlo a tu Python. Vamos a cargar también esta función para escalar datos, esta para hacer validación externa, una sola, y para evaluar nuestro modelo. Esta es nuestra base de datos. Seleccionamos todos los datos para que no haya datos faltantes en la columna respuesta. Seleccionamos 10 000 datos. Y aquí te dejo el código, por si quieres usar variables categóricas y convertirlas en "dummies". Aquí selecciono las variables que usaré como regresoras y la variable respuesta. Y creo las muestras: 'train' y 'test'. Aquí ya he escalado los datos. Recuerda que para…

Contenido