Update README.md
Browse files
README.md
CHANGED
@@ -9,7 +9,7 @@ tags:
|
|
9 |
|
10 |
# mchochlov/codebert-base-cd-ft
|
11 |
|
12 |
-
This is a [sentence-transformers](https://www.SBERT.net) model: It maps
|
13 |
|
14 |
<!--- Describe your model here -->
|
15 |
|
@@ -25,10 +25,10 @@ Then you can use the model like this:
|
|
25 |
|
26 |
```python
|
27 |
from sentence_transformers import SentenceTransformer
|
28 |
-
|
29 |
|
30 |
model = SentenceTransformer('mchochlov/codebert-base-cd-ft')
|
31 |
-
embeddings = model.encode(
|
32 |
print(embeddings)
|
33 |
```
|
34 |
|
|
|
9 |
|
10 |
# mchochlov/codebert-base-cd-ft
|
11 |
|
12 |
+
This is a [sentence-transformers](https://www.SBERT.net) model: It maps code to a 768 dimensional dense vector space and is specifically fine tuned towards clone detection using contrastive learning on parts of BigCloneBench code.
|
13 |
|
14 |
<!--- Describe your model here -->
|
15 |
|
|
|
25 |
|
26 |
```python
|
27 |
from sentence_transformers import SentenceTransformer
|
28 |
+
code_fragments = [...]
|
29 |
|
30 |
model = SentenceTransformer('mchochlov/codebert-base-cd-ft')
|
31 |
+
embeddings = model.encode(code_fragments)
|
32 |
print(embeddings)
|
33 |
```
|
34 |
|