FAQ: Decision Trees - Weighted Information Gain

This community-built FAQ covers the “Weighted Information Gain” exercise from the lesson “Decision Trees”.

Paths and Courses
This exercise can be found in the following Codecademy content:

Data Science

Machine Learning

FAQs on the exercise Weighted Information Gain

There are currently no frequently asked questions associated with this exercise – that’s where you come in! You can contribute to this section by offering your own questions, answers, or clarifications on this exercise. Ask or answer a question by clicking reply (reply) below.

If you’ve had an “aha” moment about the concepts, formatting, syntax, or anything else with this exercise, consider sharing those insights! Teaching others and answering their questions is one of the best ways to learn and stay sharp.

Join the Discussion. Help a fellow learner on their journey.

Ask or answer a question about this exercise by clicking reply (reply) below!

Agree with a comment or answer? Like (like) to up-vote the contribution!

Need broader help or resources? Head here.

Looking for motivation to keep learning? Join our wider discussions.

Learn more about how to use this guide.

Found a bug? Report it!

Have a question about your account or billing? Reach out to our customer support team!

None of the above? Find out where to ask other questions here!

How is defined the quantity of splitted sets. In the lesson’ example it is 3. Though, in the function split(cars, car_labels, 3) - 3 is the index of the column and not the quantity of sets.

Hi there!
I have troubles understanding understanding the split function given in this exercise (although it isn’t explained anywhere).

def split(dataset, labels, column):
    data_subsets = []
    label_subsets = []
    counts = list(set([data[column] for data in dataset]))
    for k in counts:
        new_data_subset = []
        new_label_subset = []
        for i in range(len(dataset)):
            if dataset[i][column] == k:
    return data_subsets, label_subsets

Could someone explain the code to me?

Thanks in advance!

me too. ```
list(set([data[column] for data in dataset]))

what does set do here eactly