2
$\begingroup$

I have a training data set for a binary classification problem. There exist two possible scenarios, one is that all of the training data set are labeled as positive; another one is that the training data set includes labeled positive ones and labeled negative ones.

Assume that I use this training data set to train a decision tree. How do these two different scenarios affect the trained tree?

Moreover, if the ratio of positive and negative ones changes, how does this change affect the built tree model?

$\endgroup$

1 Answer 1

0
$\begingroup$

(this is my first answer in stackexchange)

There will be a huge difference! If all your training data is labelled as positive, the entropy for each attribute will be 0.

Since the information gain for an attribute is defined as 1 - entropy, your learned model will only have one arbitrary node. In that case, if you test your tree, you will only get positive results.

It is always important to have a good balance between negatives and positives in your training data sets.

Hope that helped :)

$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge that you have read and understand our privacy policy and code of conduct.

Not the answer you're looking for? Browse other questions tagged or ask your own question.