ludusc commited on
Commit
9676870
1 Parent(s): 8a2c29c
Files changed (1) hide show
  1. pages/2_Concepts_comparison.py +6 -6
pages/2_Concepts_comparison.py CHANGED
@@ -159,13 +159,13 @@ with output_col_2:
159
  header_col_2.write(f'Concepts {", ".join(concept_ids)} - Latent space CLIP')# - Nodes {",".join(list(imp_nodes))}')
160
 
161
  edges = []
162
- for i in range(len(concept_ids)):
163
- for j in range(len(concept_ids)):
164
- if i != j:
165
- print(f'Similarity between {concept_ids[i]} and {concept_ids[j]}')
166
- similarity = cosine_similarity(vectors[i,:].reshape(1, -1), vectors[j,:].reshape(1, -1))
167
  print(np.round(similarity[0][0], 3))
168
- edges.append((concept_ids[i], concept_ids[j], np.round(similarity[0][0], 3)))
169
 
170
  # # Create an empty graph
171
  # G = nx.Graph()
 
159
  header_col_2.write(f'Concepts {", ".join(concept_ids)} - Latent space CLIP')# - Nodes {",".join(list(imp_nodes))}')
160
 
161
  edges = []
162
+ for c1 in concept_ids:
163
+ for c2 in concept_ids:
164
+ if c1 != c2:
165
+ print(f'Similarity between {c1} and {c2}')
166
+ similarity = cosine_similarity(vectors[c1].reshape(1, -1), vectors[c2].reshape(1, -1))
167
  print(np.round(similarity[0][0], 3))
168
+ edges.append((c1, c2, np.round(similarity[0][0], 3)))
169
 
170
  # # Create an empty graph
171
  # G = nx.Graph()