Open intMinsu opened 2 weeks ago
Thanks for your interest!
The native code supports classification only. I think with some modifications it should be able to handle regression, basically changing the BCE classification loss into a regression loss, and substituting label embedding from one-hot classes into something continuous.
Hi, thanks for opening the code. Can I use this model to solve regression problem?