Closed Innixma closed 2 years ago
Hi Nick, you're right this is currently not supported (the shrink function is written only for univariate regression/binary classification and misbehaves with multiple classes). It's a pretty straightforward extension though and we can get to it soon!
@aagarwal1996 @yanshuotan
That would be amazing, thanks!
@Innixma should work now :)
Awesome, thank you so much!
Does HSTree support multiclass classification problems with RandomForest / ExtraTrees as the estimator?
From my initial tests it appears buggy. Calling
predict_proba
with the final model results in lots of NaN predictions, along with warnings during training such as:If helpful I can try to create a reproducible example.
Here is an example result comparing with sklearn default RF (
_og_
) with accuracy metric. Because HSTree returns many NaN predictions, the scores are very low.One observation is the scores get worse the more trees there are in HSTree forests. I'd guess the likelihood of returning a NaN result is increasing with the number of trees.