Sentence Similarity
sentence-transformers
PyTorch
Transformers
bert
feature-extraction
text-embeddings-inference
Instructions to use YangsHao/RecBERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use YangsHao/RecBERT with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("YangsHao/RecBERT") sentences = [ "汪汪队立大功第1季动画动画冒险剧情本领高强的狗狗巡逻队精通科技的10岁男孩", "超人总动员2喜剧动作动画冒险家庭亲情超级英雄励志超能先生变奶爸超人家族时隔14年强势回归", "星汉灿烂·月升沧海剧情爱情星汉灿烂·月升沧海该剧讲述了程家女名少商", "外星人事件2喜剧科幻剧情山炮大战爆笑来袭传闻,几十年前外星人曾开着飞船造访过下井沟" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Transformers
How to use YangsHao/RecBERT with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("YangsHao/RecBERT") model = AutoModel.from_pretrained("YangsHao/RecBERT") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -8,9 +8,10 @@ tags:
|
|
| 8 |
|
| 9 |
---
|
| 10 |
|
| 11 |
-
#
|
|
|
|
|
|
|
| 12 |
|
| 13 |
-
This is a [sentence-transformers](https://www.SBERT.net) model: It maps sentences & paragraphs to a 768 dimensional dense vector space and can be used for tasks like clustering or semantic search.
|
| 14 |
|
| 15 |
<!--- Describe your model here -->
|
| 16 |
|
|
|
|
| 8 |
|
| 9 |
---
|
| 10 |
|
| 11 |
+
# RecBERT
|
| 12 |
+
|
| 13 |
+
RecBERT is a Pretrain-for-Recommentation model. It could serve as an item encoder with the capability of capturing both collaborative filtering signals and text semantic Information. RecBERT is pre-trained on CMCC, an industrial video recommendation dataset collected from China Mobile.
|
| 14 |
|
|
|
|
| 15 |
|
| 16 |
<!--- Describe your model here -->
|
| 17 |
|