mirror of
https://github.com/fccapria/scientify.git
synced 2026-01-12 02:36:10 +00:00
6 lines
No EOL
239 B
Python
6 lines
No EOL
239 B
Python
import yake
|
|
|
|
def extract_keywords(text: str, num_keywords: int = 5) -> list:
|
|
kw_extractor = yake.KeywordExtractor(lan="en", n=1, top=num_keywords)
|
|
keywords = kw_extractor.extract_keywords(text)
|
|
return [kw for kw, _ in keywords] |