dmlc / dmlc/treelite

Add `impurity` field in Treelite model representation

Open
#682 0 comments 0 reactions 0 assignees View on GitHub
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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.