out vector should be of size nClasses for softmax

guitargeek / XGBoost-FastForest

Minimal library code to deploy XGBoost models in C++.

MIT License

89 stars 30 forks source link

Closed svb688 closed 4 years ago

svb688 commented 4 years ago

I was running into a malloc error when I tried to requery the boosted model. I think it was due to the incorrect vector size.

guitargeek commented 4 years ago

My bad, my tests only checked the case n_classes == 3. Thanks for the PR, really appreciated!