Update README.md
Browse files
README.md
CHANGED
@@ -16,7 +16,8 @@ def cal_score(a, b):
|
|
16 |
b_norm = b / b.norm(dim=1)[:, None]
|
17 |
return torch.mm(a_norm, b_norm.transpose(0, 1)) * 100
|
18 |
|
19 |
-
model = AutoModel.from_pretrained('BM-K/KoSimCSE-bert')
|
|
|
20 |
|
21 |
sentences = ['์นํ๊ฐ ๋คํ์ ๊ฐ๋ก ์ง๋ฌ ๋จน์ด๋ฅผ ์ซ๋๋ค.',
|
22 |
'์นํ ํ ๋ง๋ฆฌ๊ฐ ๋จน์ด ๋ค์์ ๋ฌ๋ฆฌ๊ณ ์๋ค.',
|
|
|
16 |
b_norm = b / b.norm(dim=1)[:, None]
|
17 |
return torch.mm(a_norm, b_norm.transpose(0, 1)) * 100
|
18 |
|
19 |
+
model = AutoModel.from_pretrained('BM-K/KoSimCSE-bert')
|
20 |
+
AutoTokenizer.from_pretrained('BM-K/KoSimCSE-bert')
|
21 |
|
22 |
sentences = ['์นํ๊ฐ ๋คํ์ ๊ฐ๋ก ์ง๋ฌ ๋จน์ด๋ฅผ ์ซ๋๋ค.',
|
23 |
'์นํ ํ ๋ง๋ฆฌ๊ฐ ๋จน์ด ๋ค์์ ๋ฌ๋ฆฌ๊ณ ์๋ค.',
|