Construir la tabla del predictor insight graph
En el ejercicio anterior aprendiste a calcular la columna de incidencia de la tabla del predictor insight graph. En este ejercicio, también añadirás el tamaño de los grupos y encapsularás todo en una función que devuelva la tabla del predictor insight graph para una variable dada.
Este ejercicio forma parte del curso
Introducción al análisis predictivo en Python
Instrucciones del ejercicio
- Agrupa la basetable por
variable. - Calcula la tabla del predictor insight graph calculando la incidencia del target y los tamaños de los grupos.
- Usa la función
create_pig_tablepara calcular la tabla del predictor insight graph para la variable "gender".
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Function that creates predictor insight graph table
def create_pig_table(basetable, target, variable):
# Create groups for each variable
groups = basetable[[target,variable]].____(____)
# Calculate size and target incidence for each group
pig_table = groups[____].agg(Incidence = '____', Size = '____').reset_index()
# Return the predictor insight graph table
return pig_table
# Calculate the predictor insight graph table for the variable gender
pig_table_gender = ____(basetable, "target", ____)
# Print the result
print(pig_table_gender)