cerlymarco / linear-tree

A python library to build Model Trees with Linear Models at the leaves.
MIT License
338 stars 54 forks source link

Linear Boosting will it work for categorical features? #27

Closed Sandy4321 closed 1 year ago

Sandy4321 commented 1 year ago

Linear Boosting will it work for categorical features?

cerlymarco commented 1 year ago

Hi,

ALL the estimators in linear-tree are sklearn estiamtors. They don't automatically encode categorical features. You should use your own strategy for categorical features encoding

If you support support the project, don't forget to leave a star ;-)

Sandy4321 commented 1 year ago

Star done

Sandy4321 commented 1 year ago

Do you have examples When your code is much better than gxboost? Do you planning to add your linear regression splits to gradient busting? Seems to be some of gbm packages already has linear regression splits?