Trazar los tamaños de los grupos
El predictor insight graph ofrece información sobre las variables predictoras. Cada variable divide a la población en varios grupos. El predictor insight graph muestra una línea con la incidencia media del target para cada grupo y unas barras con los tamaños de los grupos. En este ejercicio, aprenderás a escribir y aplicar una función que trace un predictor insight graph a partir de una tabla de predictor insight graph.
Este ejercicio forma parte del curso
Introducción al análisis predictivo en Python
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# The function to plot a predictor insight graph
def plot_pig(pig_table, variable):
# Plot formatting
plt.ylabel("Size", rotation=0, rotation_mode="anchor", ha="right")
# Plot the bars with sizes
pig_table["____"].plot(kind="bar", width=0.5, color="lightgray", edgecolor="none")
# Plot the incidence line on secondary axis
pig_table["____"].plot(secondary_y=True)
# Plot formatting
plt.xticks(np.arange(len(pig_table)), pig_table[variable])
plt.xlim([-0.5, len(pig_table) - 0.5])
plt.ylabel("Incidence", rotation=0, rotation_mode="anchor", ha="left")
# Show the graph
plt.show()