utterworks / fast-bert

Super easy library for BERT based NLP models
Apache License 2.0
1.86k stars 341 forks source link

How to do regression analysis with fast-bert? #275

Open bfsujason opened 3 years ago

bfsujason commented 3 years ago

I want to use fast-bert to process my dataset with the labels being numerical values instead of categories.

Does fast-bert provide an option for doing regression analysis with numerical values?

If not, which part of the code should I hack into for regression task?

Many thanks!