De groepsgroottes plotten

De predictor insight graph geeft informatie over voorspellende variabelen. Elke variabele verdeelt de populatie in meerdere groepen. De predictor insight graph bevat een lijn die de gemiddelde target-incidentie per groep toont en een balk die de groepsgroottes weergeeft. In deze oefening leer je een functie schrijven en gebruiken die een predictor insight graph plot op basis van een predictor-insightgraphtabel.

Deze oefening maakt deel uit van de cursus

Introductie tot Predictive Analytics in Python

Bekijk cursus

Interactieve oefening met praktijkervaring

Probeer deze oefening door deze voorbeeldcode aan te vullen.

# 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()

Code bewerken en uitvoeren