# ---
# jupyter:
#   jupytext:
#     text_representation:
#       extension: .py
#       format_name: light
#       format_version: '1.5'
#       jupytext_version: 1.14.4
#   kernelspec:
#     display_name: kgtk_env
#     language: python
#     name: kgtk_env
# ---

from cosine_similarity import CosineSimilarity
import json
import pandas as pd
from IPython.display import display, HTML

v_path = '/Users/amandeep/Github/maa-analysis/MAA_Datasets/v3.2.0'

cs = CosineSimilarity(f'{v_path}/text_embeddings_all.tsv', f'{v_path}/qnodes-properties-labels-for-V3.2.0_KB.tsv')

r = []
r.append(cs.compute_similarity('Q37828', 'Q271997'))
r.append(cs.compute_similarity('Q48989064', 'Q127956'))
r.append(cs.compute_similarity('Q271997', 'Q159683'))
r.append(cs.compute_similarity('Q48989064', 'Q271997'))
df = pd.DataFrame(r)
display(HTML(df.to_html()))



StarcoderdataJupyterScriptsDedupFiltered