search

cerlymarco / linear-tree

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

Extract Coefficients #5

Closed YorgosMoiragias closed 2 years ago

YorgosMoiragias commented 2 years ago

How can someone extract the coefficients of each linear model implemented in each leaf?

cerlymarco commented 2 years ago

https://towardsdatascience.com/explainable-ai-with-linear-trees-7e30a6f067d7

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

unnamed-idea commented 2 years ago

hello, your link does not include the code to get the coefficients of the each of the leaf estimators (eg linear regression). Can you elaborate on this subject?

cerlymarco commented 2 years ago

At the bottom of the post, click to "CHECK MY GITHUB REPOSITORY".

The notebook you are interested in is: https://github.com/cerlymarco/MEDIUM_NoteBook/tree/master/ModelTrees_Explainability

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