forestry-labs / Rforestry

https://forestry-labs.github.io/Rforestry/
34 stars 10 forks source link

Python: RandomForest conforms to sklearn Estimator interface #146

Open Ilia-Shutov opened 1 year ago

Ilia-Shutov commented 1 year ago

Note: reconstruct_forest (reconstructForest) deserialization logic is changed as well as Python part of serialization: tree_counts contains 3 values for each tree instead of 4: numNodes, numSplitIdx, numAvIdx Verified by adding test_pickle_unpickle.py, also sklearn test set contains its own pickle - unpickle test

Issue https://github.com/forestry-labs/Rforestry/issues/98

Continuation of https://github.com/forestry-labs/Rforestry/pull/101 (closed in favour of this one)