mlr-org / mlr3pipelines

Dataflow Programming for Machine Learning in R
https://mlr3pipelines.mlr-org.com/
GNU Lesser General Public License v3.0
137 stars 25 forks source link

regression tree based binning of cts features #813

Open mb706 opened 3 weeks ago

mb706 commented 3 weeks ago

https://github.com/yandex-research/rtdl-num-embeddings/blob/abf8a8b35854e4b06476bb48902096b0b58ffce2/package/rtdl_num_embeddings.py#L300

https://arxiv.org/abs/2203.05556#

Could either be its own pipeop or an extension of an existing binning pipeop