Update README.md
Browse files
README.md
CHANGED
@@ -7,12 +7,17 @@ tags:
|
|
7 |
- sentence-transformers
|
8 |
- sentence-similarity
|
9 |
- feature-extraction
|
|
|
10 |
widget: []
|
11 |
---
|
12 |
|
13 |
-
#
|
14 |
|
15 |
-
|
|
|
|
|
|
|
|
|
16 |
|
17 |
## Model Details
|
18 |
|
@@ -26,11 +31,6 @@ This is a [sentence-transformers](https://www.SBERT.net) model trained. It maps
|
|
26 |
<!-- - **Language:** Unknown -->
|
27 |
<!-- - **License:** Unknown -->
|
28 |
|
29 |
-
### Model Sources
|
30 |
-
|
31 |
-
- **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
|
32 |
-
- **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
|
33 |
-
- **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
|
34 |
|
35 |
### Full Model Architecture
|
36 |
|
@@ -42,74 +42,6 @@ SentenceTransformer(
|
|
42 |
)
|
43 |
```
|
44 |
|
45 |
-
## Usage
|
46 |
-
|
47 |
-
### Direct Usage (Sentence Transformers)
|
48 |
-
|
49 |
-
First install the Sentence Transformers library:
|
50 |
-
|
51 |
-
```bash
|
52 |
-
pip install -U sentence-transformers
|
53 |
-
```
|
54 |
-
|
55 |
-
Then you can load this model and run inference.
|
56 |
-
```python
|
57 |
-
from sentence_transformers import SentenceTransformer
|
58 |
-
|
59 |
-
# Download from the 🤗 Hub
|
60 |
-
model = SentenceTransformer("ZhishanQ/UniHGKR-base")
|
61 |
-
# Run inference
|
62 |
-
sentences = [
|
63 |
-
'The weather is lovely today.',
|
64 |
-
"It's so sunny outside!",
|
65 |
-
'He drove to the stadium.',
|
66 |
-
]
|
67 |
-
embeddings = model.encode(sentences)
|
68 |
-
print(embeddings.shape)
|
69 |
-
# [3, 768]
|
70 |
-
|
71 |
-
# Get the similarity scores for the embeddings
|
72 |
-
similarities = model.similarity(embeddings, embeddings)
|
73 |
-
print(similarities.shape)
|
74 |
-
# [3, 3]
|
75 |
-
```
|
76 |
-
|
77 |
-
<!--
|
78 |
-
### Direct Usage (Transformers)
|
79 |
-
|
80 |
-
<details><summary>Click to see the direct usage in Transformers</summary>
|
81 |
-
|
82 |
-
</details>
|
83 |
-
-->
|
84 |
-
|
85 |
-
<!--
|
86 |
-
### Downstream Usage (Sentence Transformers)
|
87 |
-
|
88 |
-
You can finetune this model on your own dataset.
|
89 |
-
|
90 |
-
<details><summary>Click to expand</summary>
|
91 |
-
|
92 |
-
</details>
|
93 |
-
-->
|
94 |
-
|
95 |
-
<!--
|
96 |
-
### Out-of-Scope Use
|
97 |
-
|
98 |
-
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
99 |
-
-->
|
100 |
-
|
101 |
-
<!--
|
102 |
-
## Bias, Risks and Limitations
|
103 |
-
|
104 |
-
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
105 |
-
-->
|
106 |
-
|
107 |
-
<!--
|
108 |
-
### Recommendations
|
109 |
-
|
110 |
-
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
111 |
-
-->
|
112 |
-
|
113 |
## Training Details
|
114 |
|
115 |
### Framework Versions
|
|
|
7 |
- sentence-transformers
|
8 |
- sentence-similarity
|
9 |
- feature-extraction
|
10 |
+
- UniHGKR-base
|
11 |
widget: []
|
12 |
---
|
13 |
|
14 |
+
# UniHGKR-base
|
15 |
|
16 |
+
Please see github repository [UniHGKR](https://github.com/ZhishanQ/UniHGKR/tree/main/code_for_UniHGKR_base) to know how to use this model.
|
17 |
+
|
18 |
+
We recommend using the [sentence-transformers](https://www.SBERT.net) package to load our model and to perform embedding for paragraphs and sentences.
|
19 |
+
|
20 |
+
It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
|
21 |
|
22 |
## Model Details
|
23 |
|
|
|
31 |
<!-- - **Language:** Unknown -->
|
32 |
<!-- - **License:** Unknown -->
|
33 |
|
|
|
|
|
|
|
|
|
|
|
34 |
|
35 |
### Full Model Architecture
|
36 |
|
|
|
42 |
)
|
43 |
```
|
44 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
45 |
## Training Details
|
46 |
|
47 |
### Framework Versions
|