Constructing the predictor insight graph table
In the previous exercise you learned how to calculate the incidence column of the predictor insight graph table. In this exercise, you will also add the size of the groups, and wrap everything in a function that returns the predictor insight graph table for a given variable.
This exercise is part of the course
Introduction to Predictive Analytics in Python
Exercise instructions
- Group the basetable by
variable
. - Calculate the predictor insight graph table by calculating the target incidence and group sizes.
- Use the function
create_pig_table
to calculate the predictor insight graph table for the variable "gender".
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# 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)