Test_Bot / rake_find.py
aguyader's picture
Upload folder using huggingface_hub
f307bb6 verified
raw
history blame contribute delete
453 Bytes
from rake_nltk import Rake
import nltk
nltk.download('punkt_tab')
def extract_keywords_rake(question):
r = Rake()
# Extract keywords
r.extract_keywords_from_text(question)
keywords = r.get_ranked_phrases()
return keywords
# Example chatbot question
question = "What is the weather like in New York City today?"
# Extract keywords using Rake
keywords_rake = extract_keywords_rake(question)
print("Keywords:", keywords_rake)