tuhailong commited on
Commit
4f41d02
2 Parent(s): 24888f0 37f7a4d

Merge branch 'main' of https://huggingface.co/tuhailong/simcse_model into main

Browse files
Files changed (1) hide show
  1. README.md +23 -0
README.md ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: zh
3
+ tags:
4
+ - simcse
5
+ datasets:
6
+ - dialogue
7
+ ---
8
+
9
+ # Data
10
+ unsupervise train data is E-commerce dialogue.
11
+
12
+ ## Model
13
+ model is [simcse](https://arxiv.org/abs/2104.08821).
14
+
15
+ ### Usage
16
+ ```python
17
+ >>> from transformers import AutoTokenizer, AutoModel
18
+ >>> model = AutoModel.from_pretrained("tuhailong/simcse_model")
19
+ >>> tokenizer = AutoTokenizer.from_pretrained("tuhailong/simcse_model")
20
+ >>> sentences_str_list = ["今天天气不错的","天气不错的"]
21
+ >>> inputs = tokenizer(sentences_str_list,return_tensors="pt", padding='max_length', truncation=True, max_length=32)
22
+ >>> outputs = model(**inputs)
23
+ ```