search

shon-otmazgin / fastcoref

MIT License
142 stars 25 forks source link

Add custom nlp for CorefTrainer #36

Closed wannaphong closed 1 year ago

wannaphong commented 1 year ago

Add custom nlp for CorefTrainer to train more language.

shon-otmazgin commented 1 year ago

Hi The trainer must have nlp object, and the use of tokenized text in predict function is only possible for inference, can you change the if/else statements accordingly?

wannaphong commented 1 year ago

Hi The trainer must have nlp object, and the use of tokenized text in predict function is only possible for inference, can you change the if/else statements accordingly?

OK. I was change the if/else statements.

shon-otmazgin commented 1 year ago

Thanks!. lastly, also needs to update the README in the train section since you added a new parameter to the trainer class

shon-otmazgin commented 1 year ago

@wannaphong I just released version 2.1.6 thanks.