pedroilidio / bipartite_learn

BSD 3-Clause "New" or "Revised" License
3 stars 1 forks source link

REF PERF Inference methods GBM optimized to bipartite data #11

Open pedroilidio opened 2 days ago

pedroilidio commented 2 days ago

Currently, GBMs convert bipartite feature matrices to the SGSO format before predicting, requiring minimal code changes over the sklearn implementation. However, inference could go from $n_\text{pred}^2 \log (n_\text{train})$ to $n_\text{pred} \log (n_\text{train})$ with more clever code design (see the bipartite apply function in tree/_tree.pyx).