Spaces:
Runtime error
Runtime error
File size: 723 Bytes
d0de866 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from rake_nltk import Rake as RAKE
class Rake():
def __init__(self):
self.rake = RAKE()
def get_keyphrases_with_score(self, rake_input):
if isinstance(rake_input, list):
self.rake.extract_keywords_from_sentences(rake_input)
elif isinstance(rake_input, str):
self.rake.extract_keywords_from_text(rake_input)
return self.rake.get_ranked_phrases_with_scores()
def get_keyphrases(self, rake_input):
if isinstance(rake_input, list):
self.rake.extract_keywords_from_sentences(rake_input)
elif isinstance(rake_input, str):
self.rake.extract_keywords_from_text(rake_input)
return self.rake.get_ranked_phrases()
|