search

automl / TabPFN

Official implementation of the TabPFN paper (https://arxiv.org/abs/2207.01848) and the tabpfn package.
http://priorlabs.ai
Apache License 2.0
1.22k stars 109 forks source link

Can we use TabPFN for regression tasks? #22

Closed mark0935git closed 1 year ago

mark0935git commented 1 year ago

Hi TabPFN developer,

Thanks for this amazing algorithm! The algorithm works well for classification tasks.

But I am wondering how we modify the script to perform regression tasks?

I assume we only need to change the output layer and loss function. Could you please some instructions.

Many thanks in advance!

noahho commented 1 year ago

Hi mark0935git,

We are currently working on that and hope to be able to publish these models soon, we'll send an update as soon as it is released!

Best Noah

Borda commented 1 year ago

We are currently working on that and hope to be able to publish these models soon, we'll send an update as soon as it is released!

hello, may I kindly ask if there was any update in the regressor direction? :rabbit: