README.md

metadata

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，model struct is cross-encoder

Usage

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

Code

train code from https://github.com/TTurn/cross-encoder