Sentence Similarity
sentence-transformers
Safetensors
Transformers
ONNX
bert
feature-extraction
text-embeddings-inference
Instructions to use JayThinkDiff/CRE-0.5 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use JayThinkDiff/CRE-0.5 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("JayThinkDiff/CRE-0.5") 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 JayThinkDiff/CRE-0.5 with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("JayThinkDiff/CRE-0.5") model = AutoModel.from_pretrained("JayThinkDiff/CRE-0.5") - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse files- 1_Pooling/config.json +10 -0
- README.md +89 -3
- config.json +40 -0
- config_sentence_transformers.json +10 -0
- model.safetensors +3 -0
- modules.json +14 -0
- sentence_bert_config.json +4 -0
- special_tokens_map.json +37 -0
- tokenizer.json +0 -0
- tokenizer_config.json +58 -0
- vocab.txt +0 -0
1_Pooling/config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"word_embedding_dimension": 1024,
|
| 3 |
+
"pooling_mode_cls_token": true,
|
| 4 |
+
"pooling_mode_mean_tokens": false,
|
| 5 |
+
"pooling_mode_max_tokens": false,
|
| 6 |
+
"pooling_mode_mean_sqrt_len_tokens": false,
|
| 7 |
+
"pooling_mode_weightedmean_tokens": false,
|
| 8 |
+
"pooling_mode_lasttoken": false,
|
| 9 |
+
"include_prompt": true
|
| 10 |
+
}
|
README.md
CHANGED
|
@@ -1,3 +1,89 @@
|
|
| 1 |
-
---
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
pipeline_tag: sentence-similarity
|
| 3 |
+
tags:
|
| 4 |
+
- sentence-transformers
|
| 5 |
+
- feature-extraction
|
| 6 |
+
- sentence-similarity
|
| 7 |
+
- transformers
|
| 8 |
+
- onnx
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
CRE(CareerInternational Recruitment Embedding)是一个工作技能和招聘的预训练语言模型。
|
| 12 |
+
|
| 13 |
+
<small>
|
| 14 |
+
|
| 15 |
+
CRE的版本管理采用GNU版本号。SNAPSHOT代表开发版本,模型随时会被更新。RELEASE代表正式版本,模型之后不会再进行更新。需要在模型调用时指定Version版本。
|
| 16 |
+
|
| 17 |
+
示例:
|
| 18 |
+
|
| 19 |
+
```
|
| 20 |
+
!pip install -U sentence-transformers
|
| 21 |
+
|
| 22 |
+
from sentence_transformers import SentenceTransformer, util
|
| 23 |
+
|
| 24 |
+
access_token = "<Your HuggingFace Token>"
|
| 25 |
+
|
| 26 |
+
'''如不填写,默认为主干分支。可替换下面的revision为期望的版本号。例如:0.1.0-RELEASE'''
|
| 27 |
+
model = SentenceTransformer("CITech/CRE",revision="main",token=access_token)
|
| 28 |
+
|
| 29 |
+
query_embedding = model.encode("嵌入式")
|
| 30 |
+
passage_embedding = model.encode([
|
| 31 |
+
"岗位职责:1.从事通讯产品相关嵌入式软件研发工作;2.进行软件详细设计,代码编写,单元测试,集成测试等;3.进行软件代码的维护和改进工作;4.完成部门安排的其它研发相关工作。任职资格:1.通信,计算机,电子,自动化等相关专业本科及以上学历,英语CET-4以上,具备英文技术资料阅读能力;2.熟练掌握C语言程序设计,熟悉软件开发过程;4.有数通领域(交换/路由协议)开发经验者优先;有TCP/IP栈,路由协议/MPLS协议等开发经验者优先;有BROADCOM/MARVELL/INTEL系列多核处理器/转发芯片/网络处理器/交换芯片等开发经验者优先;熟悉软件架构和软件流程,有过大型嵌入式软件或平台软件设计方面经验者优先。5. 具有独立思考和自我学习能力;拥有良好的工作态度和服务敬业精神;积极上进,具有团队合作精神;沟通表达能力强,能适应加班和出差",
|
| 32 |
+
"招聘嵌入式系统工程师,要求会 设计嵌入式系统及单片机、会软件编程!PCB设计:AD、Pulsonix、Cadence(至少会一种)编程语言要求会:C、C++ 、Java、Python (至少会两种,Python必须会)3D设计要求会:CATIA 、SOLIDWORKS、 AutoCAD (至少会一种)工作内容:设计、开发嵌入式系统;构造嵌入式系统的框架结构、内核原理;负责编写整体系统设计方案;负责嵌入式硬件、软件开发工作;对客户进行系统技术支持。工作地点:山西晋中薪酬待遇:依据要求面谈,公司利润分红!",
|
| 33 |
+
])
|
| 34 |
+
|
| 35 |
+
print("查询结果:", util.cos_sim(query_embedding, passage_embedding))
|
| 36 |
+
```
|
| 37 |
+
|
| 38 |
+
<ul>注意事项:
|
| 39 |
+
<li>使用CLS Token来表征句子</li>
|
| 40 |
+
<li>最大输入Token长度为512</li>
|
| 41 |
+
</ul>
|
| 42 |
+
</small>
|
| 43 |
+
|
| 44 |
+
---
|
| 45 |
+
更新日志:
|
| 46 |
+
<small>
|
| 47 |
+
|
| 48 |
+
|
| 49 |
+
<b>0.1.0-RELEASE 2024/04/02</b>
|
| 50 |
+
<ul>
|
| 51 |
+
<li>新增:</li>
|
| 52 |
+
<ul>
|
| 53 |
+
<li><strong>模型微调</strong>:引入基于智源(BAAI)bge-large-zh-v1.5模型的微调版本,作为项目的基础模型。</li>
|
| 54 |
+
<li><strong>大规模训练</strong>:在32张16GB显存的NVIDIA V100 GPUs上,通过DeepSpeed技术,对2000万条经过清洗和去重职位描述(JD)数据进行了持续预训练(Continue PreTrainning)。</li>
|
| 55 |
+
<li><strong>检索预训练方法</strong>:采用RetroMAE(Retrieval-oriented Masked Auto-Encoder)算法作为句子级别的密集检索预训练方法,通过在句子级别上应用Masked Language Modeling(MLM)任务,同时结合检索机制,优化了模型对语义信息的编码能力,提升了模型的检索能力使其在处理复杂查询时更加精准和高效。</li>
|
| 56 |
+
</ul>
|
| 57 |
+
<li>改进:无</li>
|
| 58 |
+
<li>删除:无</li>
|
| 59 |
+
<li>其他:</li>
|
| 60 |
+
<ul>
|
| 61 |
+
<li><strong>训练恢复</strong>:支持从先前保存的checkpoint恢复模型训练,提高训练过程的灵活性和效率。</li>
|
| 62 |
+
<li><strong>内存优化</strong>:引入Gradient Accumulation技术,优化了模型训练过程中的内存使用效率,允许在有限的硬件资源下进行更大规模的训练。</li>
|
| 63 |
+
</ul>
|
| 64 |
+
</ul>
|
| 65 |
+
|
| 66 |
+
<b>0.2.0-RELEASE 2024/04/13</b>
|
| 67 |
+
<ul>
|
| 68 |
+
<li>新增:</li>
|
| 69 |
+
<ul>
|
| 70 |
+
<li>对职位名称、简历中的工作经历和项目经验这三种数据进行继续训练</li>
|
| 71 |
+
</ul>
|
| 72 |
+
<li>改进:
|
| 73 |
+
<ul>
|
| 74 |
+
<li>改进CLS Token的句子表征能力</li>
|
| 75 |
+
<li>模型训练精度重新调整到FP32</li>
|
| 76 |
+
<li>采用SafeTensor</li>
|
| 77 |
+
</ul>
|
| 78 |
+
</li>
|
| 79 |
+
<li>删除:无</li>
|
| 80 |
+
<li>其他:
|
| 81 |
+
<ul>
|
| 82 |
+
<li>招聘领域的指标评估体系</li>
|
| 83 |
+
<li>自动化超参选择</li>
|
| 84 |
+
</ul>
|
| 85 |
+
</li>
|
| 86 |
+
</ul>
|
| 87 |
+
|
| 88 |
+
|
| 89 |
+
</small>
|
config.json
ADDED
|
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_name_or_path": "/data1/alg/huggingface/hub/CRE_v0.3.1",
|
| 3 |
+
"architectures": [
|
| 4 |
+
"BertModel"
|
| 5 |
+
],
|
| 6 |
+
"attention_probs_dropout_prob": 0.1,
|
| 7 |
+
"bos_token_id": 0,
|
| 8 |
+
"classifier_dropout": null,
|
| 9 |
+
"directionality": "bidi",
|
| 10 |
+
"eos_token_id": 2,
|
| 11 |
+
"hidden_act": "gelu",
|
| 12 |
+
"hidden_dropout_prob": 0.1,
|
| 13 |
+
"hidden_size": 1024,
|
| 14 |
+
"id2label": {
|
| 15 |
+
"0": "LABEL_0"
|
| 16 |
+
},
|
| 17 |
+
"initializer_range": 0.02,
|
| 18 |
+
"intermediate_size": 4096,
|
| 19 |
+
"label2id": {
|
| 20 |
+
"LABEL_0": 0
|
| 21 |
+
},
|
| 22 |
+
"layer_norm_eps": 1e-12,
|
| 23 |
+
"max_position_embeddings": 512,
|
| 24 |
+
"model_type": "bert",
|
| 25 |
+
"num_attention_heads": 16,
|
| 26 |
+
"num_hidden_layers": 24,
|
| 27 |
+
"output_past": true,
|
| 28 |
+
"pad_token_id": 0,
|
| 29 |
+
"pooler_fc_size": 768,
|
| 30 |
+
"pooler_num_attention_heads": 12,
|
| 31 |
+
"pooler_num_fc_layers": 3,
|
| 32 |
+
"pooler_size_per_head": 128,
|
| 33 |
+
"pooler_type": "first_token_transform",
|
| 34 |
+
"position_embedding_type": "absolute",
|
| 35 |
+
"torch_dtype": "float32",
|
| 36 |
+
"transformers_version": "4.48.2",
|
| 37 |
+
"type_vocab_size": 2,
|
| 38 |
+
"use_cache": true,
|
| 39 |
+
"vocab_size": 21128
|
| 40 |
+
}
|
config_sentence_transformers.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"__version__": {
|
| 3 |
+
"sentence_transformers": "3.0.1",
|
| 4 |
+
"transformers": "4.48.2",
|
| 5 |
+
"pytorch": "2.3.0"
|
| 6 |
+
},
|
| 7 |
+
"prompts": {},
|
| 8 |
+
"default_prompt_name": null,
|
| 9 |
+
"similarity_fn_name": null
|
| 10 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6cfa92e1a2abd50c70a8b9652959abdc8aff20ab35935252837200c2bce21f7a
|
| 3 |
+
size 1302134568
|
modules.json
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"idx": 0,
|
| 4 |
+
"name": "0",
|
| 5 |
+
"path": "",
|
| 6 |
+
"type": "sentence_transformers.models.Transformer"
|
| 7 |
+
},
|
| 8 |
+
{
|
| 9 |
+
"idx": 1,
|
| 10 |
+
"name": "1",
|
| 11 |
+
"path": "1_Pooling",
|
| 12 |
+
"type": "sentence_transformers.models.Pooling"
|
| 13 |
+
}
|
| 14 |
+
]
|
sentence_bert_config.json
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"max_seq_length": 512,
|
| 3 |
+
"do_lower_case": false
|
| 4 |
+
}
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cls_token": {
|
| 3 |
+
"content": "[CLS]",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"mask_token": {
|
| 10 |
+
"content": "[MASK]",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
},
|
| 16 |
+
"pad_token": {
|
| 17 |
+
"content": "[PAD]",
|
| 18 |
+
"lstrip": false,
|
| 19 |
+
"normalized": false,
|
| 20 |
+
"rstrip": false,
|
| 21 |
+
"single_word": false
|
| 22 |
+
},
|
| 23 |
+
"sep_token": {
|
| 24 |
+
"content": "[SEP]",
|
| 25 |
+
"lstrip": false,
|
| 26 |
+
"normalized": false,
|
| 27 |
+
"rstrip": false,
|
| 28 |
+
"single_word": false
|
| 29 |
+
},
|
| 30 |
+
"unk_token": {
|
| 31 |
+
"content": "[UNK]",
|
| 32 |
+
"lstrip": false,
|
| 33 |
+
"normalized": false,
|
| 34 |
+
"rstrip": false,
|
| 35 |
+
"single_word": false
|
| 36 |
+
}
|
| 37 |
+
}
|
tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,58 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"added_tokens_decoder": {
|
| 3 |
+
"0": {
|
| 4 |
+
"content": "[PAD]",
|
| 5 |
+
"lstrip": false,
|
| 6 |
+
"normalized": false,
|
| 7 |
+
"rstrip": false,
|
| 8 |
+
"single_word": false,
|
| 9 |
+
"special": true
|
| 10 |
+
},
|
| 11 |
+
"100": {
|
| 12 |
+
"content": "[UNK]",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false,
|
| 17 |
+
"special": true
|
| 18 |
+
},
|
| 19 |
+
"101": {
|
| 20 |
+
"content": "[CLS]",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false,
|
| 25 |
+
"special": true
|
| 26 |
+
},
|
| 27 |
+
"102": {
|
| 28 |
+
"content": "[SEP]",
|
| 29 |
+
"lstrip": false,
|
| 30 |
+
"normalized": false,
|
| 31 |
+
"rstrip": false,
|
| 32 |
+
"single_word": false,
|
| 33 |
+
"special": true
|
| 34 |
+
},
|
| 35 |
+
"103": {
|
| 36 |
+
"content": "[MASK]",
|
| 37 |
+
"lstrip": false,
|
| 38 |
+
"normalized": false,
|
| 39 |
+
"rstrip": false,
|
| 40 |
+
"single_word": false,
|
| 41 |
+
"special": true
|
| 42 |
+
}
|
| 43 |
+
},
|
| 44 |
+
"clean_up_tokenization_spaces": true,
|
| 45 |
+
"cls_token": "[CLS]",
|
| 46 |
+
"do_basic_tokenize": true,
|
| 47 |
+
"do_lower_case": true,
|
| 48 |
+
"extra_special_tokens": {},
|
| 49 |
+
"mask_token": "[MASK]",
|
| 50 |
+
"model_max_length": 512,
|
| 51 |
+
"never_split": null,
|
| 52 |
+
"pad_token": "[PAD]",
|
| 53 |
+
"sep_token": "[SEP]",
|
| 54 |
+
"strip_accents": null,
|
| 55 |
+
"tokenize_chinese_chars": true,
|
| 56 |
+
"tokenizer_class": "BertTokenizer",
|
| 57 |
+
"unk_token": "[UNK]"
|
| 58 |
+
}
|
vocab.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|