96 lines
3.5 KiB
Python
96 lines
3.5 KiB
Python
from typing import Any
|
|
|
|
import concurrent.futures
|
|
|
|
import nltk
|
|
import numpy as np
|
|
import spacy
|
|
from nltk.corpus import stopwords
|
|
from nltk.stem import WordNetLemmatizer
|
|
from nltk.tokenize import word_tokenize
|
|
|
|
# from scipy.spatial.distance import jaccard
|
|
from sklearn.feature_extraction.text import TfidfVectorizer
|
|
|
|
nltk.download("punkt")
|
|
nltk.download("stopwords")
|
|
nltk.download("wordnet")
|
|
nltk.download("omw-1.4")
|
|
|
|
nlp = spacy.load("en_core_web_sm") # Load the English language model
|
|
lemmatizer = WordNetLemmatizer() # Initialize the WordNet lemmatizer
|
|
stop_words = set(stopwords.words("english")) # Get the English stop words
|
|
|
|
|
|
def jaccard(u: Any, v: Any) -> Any:
|
|
# Pad the shorter array with zeros at the end
|
|
u = np.pad(u, (0, max(u.shape[0], v.shape[0]) - u.shape[0]), "constant")
|
|
v = np.pad(v, (0, max(u.shape[0], v.shape[0]) - v.shape[0]), "constant")
|
|
# Calculate the Jaccard similarity
|
|
nonzero = np.bitwise_or(u != 0, v != 0)
|
|
intersection = np.bitwise_and(u != 0, v != 0)
|
|
return 1.0 - float(np.count_nonzero(intersection)) / float(
|
|
np.count_nonzero(nonzero)
|
|
)
|
|
|
|
|
|
def is_answer(sentence: str, question: str, threshold: float = 0.3) -> bool:
|
|
# Tokenize the sentence and the question
|
|
sentence_tokens = word_tokenize(sentence)
|
|
question_tokens = word_tokenize(question)
|
|
# Remove stop words from the sentence and the question
|
|
sentence_tokens = [
|
|
token for token in sentence_tokens if token.lower() not in stop_words
|
|
]
|
|
question_tokens = [
|
|
token for token in question_tokens if token.lower() not in stop_words
|
|
]
|
|
# Perform lemmatization on the sentence and the question
|
|
sentence_tokens = [lemmatizer.lemmatize(token.lower()) for token in sentence_tokens]
|
|
question_tokens = [lemmatizer.lemmatize(token.lower()) for token in question_tokens]
|
|
# Extract the main verb from the question
|
|
main_verb = None
|
|
for token in question_tokens:
|
|
if nlp(token)[0].pos_ == "VERB":
|
|
main_verb = token
|
|
break
|
|
# Generate numerical representations of the sentence and the question using TF-IDF
|
|
vectorizer = TfidfVectorizer()
|
|
sentence_vector = vectorizer.fit_transform([sentence]).toarray()[0]
|
|
question_vector = vectorizer.fit_transform([question]).toarray()[0]
|
|
# Calculate the similarity between the sentence and the question
|
|
similarity = 1 - jaccard(sentence_vector, question_vector)
|
|
# Check if the sentence answers the question
|
|
answer: bool
|
|
if main_verb is None:
|
|
answer = similarity >= threshold
|
|
return answer
|
|
else:
|
|
answer = main_verb in sentence_tokens and similarity >= threshold
|
|
return answer
|
|
|
|
|
|
# # Test the is_answer function
|
|
# sentence = "Neil Armstrong was the first person to walk on the Moon."
|
|
# question = "Who was the first person to walk on the Moon?"
|
|
# if is_answer(sentence, question):
|
|
# print("The sentence answers the question.")
|
|
# else:
|
|
# print("The sentence does not answer the question.")
|
|
|
|
# from concurrent.futures import ThreadPoolExecutor
|
|
# import concurrent.futures
|
|
|
|
|
|
def filter_irrelevant(sentences: list[str], question: str) -> list[str]:
|
|
# Create a list to store the relevant sentences
|
|
relevant_sentences = []
|
|
for sentence in sentences:
|
|
if is_answer(sentence, question):
|
|
relevant_sentences.append(sentence)
|
|
print(sentence)
|
|
return relevant_sentences
|
|
|
|
|
|
# print(filter_irrelevant_(["Neil Armstrong is an American Astronaut", "Neil Armstrong is dead", "Neil Armstrng is fake"], "Who is Neil Armstrong?"))
|