search

Shivanandroy / KeyPhraseTransformer

KeyPhraseTransformer lets you quickly extract key phrases, topics, themes from your text data with T5 transformer | Keyphrase extraction | Keyword extraction
MIT License
99 stars 13 forks source link

ways to improve performance? #2

Closed tylertaewook closed 2 years ago

tylertaewook commented 2 years ago

I came across your package through your medium post, and I'm super impressed by how good keyphrase extraction words with your model. However, I noticed it takes around 15-20s on the CPU to extract keyphrases from the 600-character text, which is suboptimal for me to use it in production project. Do you have any suggestions on how to improve performance or have any plans to publish an even lighter model?

tylertaewook commented 2 years ago

ended up being part of my mistake; never mind! your package is awesome