Add finetuning model to keytotext
gagan3012 opened this issue · comments
Gagan Bhatia commented
Is your feature request related to a problem? Please describe.
Its difficult to use it without fine-tuning on new corpus so we need to build script to finetune it on new corpus
Gagan Bhatia commented
from keytotext import KeytotextTrainer
model = KeytotextTrainer()
model.from_pretrained(model_name="t5-small")
model.train(data_df=df,batch_size=4, max_epochs=3, use_gpu=True)
model.save_model()
Gagan Bhatia commented
Trainer added