jgyasu's picture
Upload folder using huggingface_hub
436c4c1 verified
raw
history blame
5.26 kB
import re
from nltk.corpus import stopwords
def find_common_subsequences(sentence, str_list):
stop_words = set(stopwords.words('english'))
sentence = sentence.lower()
str_list = [s.lower() for s in str_list]
def is_present(subseq, str_list):
subseq_regex = re.compile(r'\b' + re.escape(subseq) + r'\b')
return all(subseq_regex.search(s) for s in str_list)
def remove_stop_words_and_special_chars(sentence):
sentence = re.sub(r'[^\w\s]', '', sentence)
words = sentence.split()
filtered_words = [word for word in words if word.lower() not in stop_words]
return " ".join(filtered_words)
cleaned_sentence = remove_stop_words_and_special_chars(sentence)
cleaned_str_list = [remove_stop_words_and_special_chars(s) for s in str_list]
words = cleaned_sentence.split()
common_grams = []
added_phrases = set()
for n in range(5, 0, -1): # Check n-grams from size 5 to 1
for i in range(len(words) - n + 1):
subseq = " ".join(words[i:i + n])
if is_present(subseq, cleaned_str_list) and not any(subseq in phrase for phrase in added_phrases):
common_grams.append((i, subseq))
added_phrases.add(subseq)
# Sort by the first appearance in the original sentence
common_grams.sort(key=lambda x: x[0])
# Assign indices based on the sorted order
indexed_common_grams = [(index + 1, subseq) for index, (_, subseq) in enumerate(common_grams)]
return indexed_common_grams
def find_common_gram_positions(str_list, common_grams):
# Initialize a list to hold positions for each sentence
positions = []
for sentence in str_list:
# Number each word in the sentence
words = re.sub(r'[^\w\s]', '', sentence).lower().split()
word_positions = {word: [] for word in words}
for idx, word in enumerate(words):
word_positions[word].append(idx + 1) # Store 1-based index positions
# Create a list to store positions of common grams for the current sentence
sentence_positions = []
for gram in common_grams:
# Clean the gram for matching
cleaned_gram = re.sub(r'[^\w\s]', '', gram).lower()
gram_words = cleaned_gram.split()
# Check for the position of the common gram in the current sentence
if all(word in word_positions for word in gram_words):
# Get the position of the first word of the common gram
start_idx = word_positions[gram_words[0]][0]
sentence_positions.append(start_idx)
else:
sentence_positions.append(-1) # Common gram not found
# Append the positions for the current sentence to the main positions list
positions.append(sentence_positions)
return positions
# # Example usage
# sentence = "Donald Trump said at a campaign rally event in Wilkes-Barre, Pennsylvania, that there has “never been a more dangerous time since the Holocaust” to be Jewish in the United States."
# str_list = [
# 'During a campaign rally in Wilkes-Barre, Pennsylvania, Donald Trump stated that being Jewish in the United States has never been more hazardous since the Holocaust.',
# 'At a campaign rally in Wilkes-Barre, Pennsylvania, Donald Trump declared that being Jewish in the United States has never been more hazardous since the Holocaust.',
# 'Donald Trump spoke at a campaign rally in Wilkes-Barre, Pennsylvania, and stated that being Jewish in the United States has never been more perilous since the Holocaust.',
# 'Donald Trump made the statement at a campaign rally in Wilkes-Barre, Pennsylvania, saying that being Jewish in the United States has never been more dangerous since the Holocaust.',
# 'Last month, Donald Trump spoke at a campaign rally in Wilkes-Barre, Pennsylvania and stated that being Jewish in the United States has never been more hazardous than during World War II.',
# 'In Wilkes-Barre, Pennsylvania, Donald Trump spoke at a campaign rally and claimed that the Holocaust was always more hazardous for Jews in the United States.',
# 'A campaign rally in Wilkes-Barre, Pennsylvania saw Donald Trump declare that being Jewish in the United States has never been more perilous since WWII.',
# 'Speaking at a campaign rally in Wilkes-Barre, Pennsylvania today, Donald Trump declared that being Jewish has never been more hazardous in the United States since the Holocaust.',
# 'During his campaign rally in Wilkes-Barre, Pennsylvania today Donald Trump stated: "There has never been a safer place for being Jewish in the United States since the Holocaust."',
# 'At a campaign rally in Wilkes-Barre, Pennsylvania (pictured), Donald Trump said, "There has never been... gotten worse for being Jewish in America since the Holocaust."'
# ]
# # Find common subsequences
# common_grams = find_common_subsequences(sentence, str_list)
# # Extract the subsequences from the common grams for position checking
# subsequences = [subseq for _, subseq in common_grams]
# # Find positions of the common grams
# common_gram_positions = find_common_gram_positions(str_list, subsequences)
# print(common_grams)