Add Doc2Vec and Sentence bert embeddings

swisscom / ai-research-keyphrase-extraction

EmbedRank: Unsupervised Keyphrase Extraction using Sentence Embeddings (official implementation)

Apache License 2.0

432 stars 88 forks source link

Add Doc2Vec and Sentence bert embeddings #35

Closed charlesBak closed 2 years ago

charlesBak commented 2 years ago

Doc2vec is added to answer the questions on how to use it. Sentence transformers is added to leverage state of the art transformers to perform keyword extraction.

Evaluation shows embedrank with sbert has the better performance.