Update README.md
Browse files
README.md
CHANGED
@@ -28,16 +28,16 @@ import torch
|
|
28 |
|
29 |
model = SentenceTransformer("AITeamVN/Vietnamese_Embedding")
|
30 |
model.max_seq_length = 2048
|
31 |
-
sentences_1 = ["Trí tuệ nhân tạo là gì", "
|
32 |
sentences_2 = ["Trí tuệ nhân tạo là công nghệ giúp máy móc suy nghĩ và học hỏi như con người. Nó hoạt động bằng cách thu thập dữ liệu, nhận diện mẫu và đưa ra quyết định.",
|
33 |
"Giấc ngủ giúp cơ thể và não bộ nghỉ ngơi, hồi phục năng lượng và cải thiện trí nhớ. Ngủ đủ giấc giúp tinh thần tỉnh táo và làm việc hiệu quả hơn."]
|
34 |
query_embedding = model.encode(sentences_1)
|
35 |
doc_embeddings = model.encode(sentences_2)
|
36 |
-
similarity = query_embedding @ doc_embeddings.T
|
37 |
-
|
|
|
|
|
38 |
'''
|
39 |
-
array([[0.6621206 , 0.33066636],
|
40 |
-
[0.18678051, 0.4875508 ]], dtype=float32)'''
|
41 |
```
|
42 |
|
43 |
|
|
|
28 |
|
29 |
model = SentenceTransformer("AITeamVN/Vietnamese_Embedding")
|
30 |
model.max_seq_length = 2048
|
31 |
+
sentences_1 = ["Trí tuệ nhân tạo là gì", "Lợi ích của giấc ngủ"]
|
32 |
sentences_2 = ["Trí tuệ nhân tạo là công nghệ giúp máy móc suy nghĩ và học hỏi như con người. Nó hoạt động bằng cách thu thập dữ liệu, nhận diện mẫu và đưa ra quyết định.",
|
33 |
"Giấc ngủ giúp cơ thể và não bộ nghỉ ngơi, hồi phục năng lượng và cải thiện trí nhớ. Ngủ đủ giấc giúp tinh thần tỉnh táo và làm việc hiệu quả hơn."]
|
34 |
query_embedding = model.encode(sentences_1)
|
35 |
doc_embeddings = model.encode(sentences_2)
|
36 |
+
print(similarity = query_embedding @ doc_embeddings.T)
|
37 |
+
'''
|
38 |
+
array([[0.66212064, 0.33066642],
|
39 |
+
[0.25866613, 0.5865289 ]], dtype=float32)
|
40 |
'''
|
|
|
|
|
41 |
```
|
42 |
|
43 |
|