README.md

metadata

language: zh
tags:
  - simcse
datasets:
  - dialogue

Data

unsupervise train data is E-commerce dialogue.

Model

model is simcse.

Usage

>>> from transformers import AutoTokenizer, AutoModel
>>> model = AutoModel.from_pretrained("tuhailong/SimCSE-bert-base")
>>> tokenizer = AutoTokenizer.from_pretrained("tuhailong/SimCSE-bert-base")
>>> sentences_str_list = ["今天天气不错的","天气不错的"]
>>> inputs = tokenizer(sentences_str_list,return_tensors="pt", padding='max_length', truncation=True, max_length=32)
>>> outputs = model(**inputs)