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 |
|