Home repository for the Regularized Greedy Forest (RGF) library. It includes original implementation from the paper and multithreaded one written in C++, along with various language-specific wrappers.
378
stars
58
forks
source link
Support latest scikit-learn (>=0.24) and Python 3.9 #333
Workaround changes in the latest scikit-learn version where integration test yields weight array with zeros:
RGF doesn't support non-positive weights: https://github.com/RGF-team/rgf/blob/8c05fd629ca478ac46b1ebed9ed4787bfa52e9c3/python-package/rgf/utils.py#L415-L420 https://github.com/RGF-team/rgf/blob/faf4b4ad6fe2d51b3ceb7384c7ac86b8071e2545/RGF/src/tet/AzTETrainer.hpp#L116-L117