Quotes Similarity Score - Medium

Quotes_similarity_score.py

co_occ_matrix = defaultdict(int)
window = 5 # not greater than 5
for sent in quotes:
    words = sent.split(' ')
    # first window 
    for i in range(0, window):
        for j in range(i+1, window):
            weight = 1/(j-i)
            ind_1 = word_to_int[words[i]]
            ind_2 = word_to_int[words[j]]
            if ind_1 == ind_2:
                continue
            co_occ_matrix[(ind_1, ind_2)] += weight
            co_occ_matrix[(ind_2, ind_1)] += weight
    # rest windows
    for i in range(window, len(words)):
        for j in range(i-window, i):
            weight = 1/(i-j)
            ind_1 = word_to_int[words[i]]
            ind_2 = word_to_int[words[j]]
            if ind_1 == ind_2:
                continue
            co_occ_matrix[(ind_1, ind_2)] += weight
            co_occ_matrix[(ind_2, ind_1)] += weight