Add `impurity` field in Treelite model representation
Open
- Dominant language
- C++
- Stars
- 830
- Forks
- 112
- Avg merge
- 1d 12h
- Merged PRs (30d)
- 18
Description
Right now, Treelite model object only keeps `gain` but discards `impurity` information from scikit-learn random forests.
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by tracing Treelite's model representation and the scikit-learn random-forest import path, focusing on where gain is currently retained and impurity is discarded. Done means impurity information from scikit-learn random forests is preserved in the Treelite model representation and existing relevant checks still pass.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp, scikit-learn
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Active
- Clarity
- Mostly clear
- Newbie friendliness
- 65/100