mim-solutions / bert_for_longer_texts

BERT classification model for processing texts longer than 512 tokens. Text is first divided into smaller chunks and after feeding them to BERT, intermediate results are pooled. The implementation allows fine-tuning.
Other
129 stars 30 forks source link

Add multiclass support #29

Closed MichalBrzozowski91 closed 5 months ago

mwachnicki commented 8 months ago

Summary of issues we have discussed today on the call:

MichalBrzozowski91 commented 7 months ago

Thanks for the review :) I added the main changes concerning division classes into Classifier and Regressor I did not implement the custom loss and optimizer:

mwachnicki commented 7 months ago

I would also love to see the same basic tests for the Regressor classes.