search

cerlymarco / linear-tree

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

[Question] Are there plans for multivariate models? #30

Closed strakehyr closed 2 years ago

cerlymarco commented 2 years ago

Hi,

the title is a little bit misleading... however, multivariate is already supported.

All the models accept 2D input features of any dimensionality like any sklearn model.

Regression models natively support multiple outputs.

Classification models only work with univariate outputs.

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