bert-extractive-summarizer icon indicating copy to clipboard operation
bert-extractive-summarizer copied to clipboard

Can we Fine-tune or Pre-train it?

Open ShoubhikBanerjee opened this issue 5 years ago • 6 comments

Is there any way to fine-tune or pre train the model.with custom dataset??

ShoubhikBanerjee avatar Jun 10 '20 12:06 ShoubhikBanerjee

You definitely can through the transformers library: https://github.com/huggingface/transformers . This library abstracts on top of this by performing different summarizations on top of embeddings.

dmmiller612 avatar Jun 11 '20 01:06 dmmiller612

I have already gone through the above link that you shared. But could not get any thing good for "extractive" summarization , fine-tune. Do you have any docs/notebook, etc.?

ShoubhikBanerjee avatar Jun 12 '20 13:06 ShoubhikBanerjee

This library abstracts on top of this by performing different summarizations on top of embeddings.

I pretrained the model with custom dataset and when passed this model as an input parameter (custom_model) am getting this error "IndexError: index -2 is out of bounds for dimension 0 with size 1". when I debug this am not getting the hiddenstates list much. plz help me on this.

nvenkatesh2409 avatar Oct 14 '20 08:10 nvenkatesh2409

please tell how to fine tune by freezing the layers and adding our own NN

abheet19 avatar Dec 19 '20 19:12 abheet19

I have already gone through the above link that you shared. But could not get any thing good for "extractive" summarization , fine-tune. Do you have any docs/notebook, etc.?

Still can not understand how to fine-tune model for "extractive" text summarizaiton task

Besides, one link I found is about abstractive summarization https://huggingface.co/course/chapter7/5?fw=tf

Any suggestion?

frankShih avatar Aug 27 '22 14:08 frankShih

Same here,

I can only find this tutorial from hugging face webpage https://huggingface.co/course/chapter7/5?fw=tf

Which is for abstractive summarization fine-tuning (if I understand correctly)

Any suggestions?

frankShih avatar Aug 29 '22 13:08 frankShih