1
$\begingroup$

I'm trying to train an XGBoost algorithm in a multiclass classification case, where the number of classes is very high (~6000). I recognize that this is quite difficult to do as it is, however, my question is about how decision trees handle this problem generally.

As such, when I split my dataset into training and test sets, there are classes that haven't been encountered before during training that we see in the test set. To me, this sounds similar to a zero-shot case, but I don't think decision trees can handle this problem.

I'm able to pass in the total number of classes (both seen and unseen), but how is this handled under the hood? Will the decision tree just assign a known class to a new test example with an unseen class?

$\endgroup$

1 Answer 1

2
$\begingroup$

Will the decision tree just assign a known class to a new test example with an unseen class?

Yes. A single decision tree assigns the majority class of the leaf to the given sample. So, if you have unseen classes in the test data, these (and possibly many) algorithms can't handle it.

$\endgroup$

Your Answer

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

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