zhichao-geng
commited on
Commit
•
36cbeb9
1
Parent(s):
1d60e18
Update README.md
Browse files
README.md
CHANGED
@@ -110,7 +110,7 @@ document_sparse_vector = get_sparse_vector(feature_document, output)
|
|
110 |
|
111 |
# get similarity score
|
112 |
sim_score = torch.matmul(query_sparse_vector[0],document_sparse_vector[0])
|
113 |
-
print(sim_score) # tensor(
|
114 |
|
115 |
|
116 |
query_token_weight = transform_sparse_vector_to_dict(query_sparse_vector)[0]
|
|
|
110 |
|
111 |
# get similarity score
|
112 |
sim_score = torch.matmul(query_sparse_vector[0],document_sparse_vector[0])
|
113 |
+
print(sim_score) # tensor(17.5307, grad_fn=<DotBackward0>)
|
114 |
|
115 |
|
116 |
query_token_weight = transform_sparse_vector_to_dict(query_sparse_vector)[0]
|