pke_en / fn.py
aka7774's picture
Upload 7 files
3086113 verified
raw
history blame contribute delete
369 Bytes
import install
import os
import pke
from spacy.lang import en
def fn_pke(text, n=10):
extractor = pke.unsupervised.MultipartiteRank()
extractor.load_document(input=text, language='en', normalization=None)
extractor.candidate_selection()
extractor.candidate_weighting()
res = extractor.get_n_best(n=n)
return "\n".join([k for k, v in res])