Sentence Similarity
sentence-transformers
PyTorch
Safetensors
Transformers
roberta
feature-extraction
text-embeddings-inference
Instructions to use mchochlov/codebert-base-cd-ft with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use mchochlov/codebert-base-cd-ft with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("mchochlov/codebert-base-cd-ft") sentences = [ "That is a happy person", "That is a happy dog", "That is a very happy person", "Today is a sunny day" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Transformers
How to use mchochlov/codebert-base-cd-ft with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("mchochlov/codebert-base-cd-ft") model = AutoModel.from_pretrained("mchochlov/codebert-base-cd-ft", device_map="auto") - Notebooks
- Google Colab
- Kaggle
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 |
|