similarity_models/cosine_similarity.py

import numpy as np
from sklearn.metrics.pairwise import cosine_similarity

class CosineSimilarity:
    def __init__(self):
        pass

    def name(self):
        return "cosine"

    def score(self,embedding_1, embedding_2):
        embedding_1 = np.array([embedding_1])
        embedding_2 = np.array([embedding_2])
        similarity_score = cosine_similarity(embedding_1, embedding_2)
        return similarity_score[0][0]