AITeamVN commited on
Commit
a5fa2bb
·
verified ·
1 Parent(s): abfb0fa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
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ì", "Tại sao giấc ngủ quan trọ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
- 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