File size: 606 Bytes
1ef9b5a a55e532 1ef9b5a a55e532 1ef9b5a a55e532 37f7a4d c5dfe30 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
---
language: zh
tags:
- simcse
datasets:
- dialogue
---
# Data
unsupervise train data is E-commerce dialogue.
## Model
model is [simcse](https://arxiv.org/abs/2104.08821).
### Usage
```python
>>> from transformers import AutoTokenizer, AutoModel
>>> model = AutoModel.from_pretrained("tuhailong/simcse_model")
>>> tokenizer = AutoTokenizer.from_pretrained("tuhailong/simcse_model")
>>> sentences_str_list = ["今天天气不错的","天气不错的"]
>>> inputs = tokenizer(sentences_str_list,return_tensors="pt", padding='max_length', truncation=True, max_length=32)
>>> outputs = model(**inputs)
``` |