Give learners "tags" like in mlr3pipelines

mlr-org / mlr3

mlr3: Machine Learning in R - next generation

https://mlr3.mlr-org.com

GNU Lesser General Public License v3.0

929 stars 86 forks source link

Give learners "tags" like in mlr3pipelines #818

Closed sebffischer closed 2 months ago

sebffischer commented 2 years ago

I think it would be a good addition to tag learners according to the category to which they belong. E.g. in mlr3extralearners we have many different implementations of very similar algorithms.

Tags I am thinking about are:

Bayesian
Boosting
Neural Network
Decision Tree
Ensemble Method
Random Forest

...

This could also be used in the learner help pages to suggest similar learners in the "seealso" section

mllg commented 2 years ago

Where are these tags stored? In the learner or the dictionary? seealso doesn't work across packages.

sebffischer commented 2 years ago

In mlr3pipelines they are stored in the PipeOp. Ok then maybe not in seealso but somewhere else in the description