Plotando os tamanhos dos grupos
O predictor insight graph traz informações sobre variáveis preditoras. Cada variável divide a população em vários grupos. O predictor insight graph tem uma linha que mostra a incidência média do alvo para cada grupo e barras que mostram os tamanhos dos grupos. Neste exercício, você vai aprender a escrever e aplicar uma função que plota um predictor insight graph, dada uma tabela de predictor insight graph.
Este exercício faz parte do curso
Introdução à Análise Preditiva em Python
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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()