search

google / yggdrasil-decision-forests

A library to train, evaluate, interpret, and productionize decision forest models such as Random Forest and Gradient Boosted Decision Trees.
https://ydf.readthedocs.io/
Apache License 2.0
498 stars 53 forks source link

feature_interaction_constraint like xgboost #129

Open marquisthunder opened 2 months ago

marquisthunder commented 2 months ago

Dear YDF team, As it is possible to specify feature interaction constraint[1] in xgboost to use domain knowledge to improve the model, is it possible to add this feature in ydf? It would be great when combined with oblique split as sometimes the oblique split condition is difficult to use in model explanation.

Referecense:

  1. https://xgboost.readthedocs.io/en/stable/tutorials/feature_interaction_constraint.html
rstz commented 2 months ago

I agree that this is a nice feature to have, we'll consider it for our roadmap! Anyone seeing this issue - Please comment or react with emoji to express support if you need this feature.