Can we use this for binary classification ?

kingfengji / mGBDT

This is the official clone for the implementation of the NIPS18 paper Multi-Layered Gradient Boosting Decision Trees (mGBDT) .

101 stars 25 forks source link

Closed OUALIDBO closed 5 years ago

OUALIDBO commented 5 years ago

If it’s possible, I want to determine the auc score. Auc_score(y,classifier.forward(y_test))

kingfengji commented 5 years ago

you can evaluate on any metrics of interest, shouldn't be hard to modify the demo script a little bit.