pytorch bert text-classification zh dataset:dialogue transformers sbert
---
language: zh
tags:
- sbert
datasets:
- dialogue
---

# Data
train data is similarity sentence data from E-commerce dialogue, about 20w sentence pairs.

## Model
model created by [sentence-tansformers](https://www.sbert.net/index.html)，model struct is cross-encoder

### Usage
```python
>>> from sentence_transformers.cross_encoder import CrossEncoder
>>> model = CrossEncoder('tuhailong/cross-encoder')
>>> scores = model.predict([["今天天气不错", "今天心情不错"]])
>>> print(scores)
```