Add "feature extraction" as our pipeline
Browse files
README.md
CHANGED
|
@@ -1,12 +1,13 @@
|
|
| 1 |
---
|
| 2 |
-
language:
|
| 3 |
-
|
| 4 |
tags:
|
| 5 |
- code
|
| 6 |
- sentence embedding
|
| 7 |
-
license:
|
| 8 |
datasets:
|
| 9 |
- CodeSearchNet
|
|
|
|
| 10 |
---
|
| 11 |
|
| 12 |
# Model Card for CodeCSE
|
|
@@ -36,4 +37,4 @@ batch = prepare_inputs(nl_json, tokenizer, args)
|
|
| 36 |
nl_inputs = batch[3]
|
| 37 |
with torch.no_grad():
|
| 38 |
nl_vec = model(input_ids=nl_inputs, sent_emb="nl")[1]
|
| 39 |
-
```
|
|
|
|
| 1 |
---
|
| 2 |
+
language:
|
| 3 |
+
- multilingual
|
| 4 |
tags:
|
| 5 |
- code
|
| 6 |
- sentence embedding
|
| 7 |
+
license: mit
|
| 8 |
datasets:
|
| 9 |
- CodeSearchNet
|
| 10 |
+
pipeline_tag: feature-extraction
|
| 11 |
---
|
| 12 |
|
| 13 |
# Model Card for CodeCSE
|
|
|
|
| 37 |
nl_inputs = batch[3]
|
| 38 |
with torch.no_grad():
|
| 39 |
nl_vec = model(input_ids=nl_inputs, sent_emb="nl")[1]
|
| 40 |
+
```
|