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 completando este código de ejemplo.
# 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()