Downloads · 30 days
8
1% of all-time downloads
cross_encoder_roberta-wwm-ext-large
tuhailong/cross_encoder_roberta-wwm-ext-large
cross_encoder_roberta-wwm-ext-large is a text classification model from tuhailong. Use it when you need a label for a piece of text. It is set up for transformers.
train data is similarity sentence data from E-commerce dialogue, about 50w sentence pairs.
All-time downloads
1K
Public
Repo size
2.6 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin1.3 GB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Apr 19, 2022
- Updated
- Apr 20, 2022
- SHA
- 7e92109a
Try a prompt
Datasets
Model card
From the Hugging Face model README
Data
train data is similarity sentence data from E-commerce dialogue, about 50w sentence pairs.
Model
model created by sentence-tansformers，model struct is cross-encoder，pretrained model is hfl/chinese-roberta-wwm-ext-large.
Code
train code from https://github.com/TTurn/cross-encoder
Usage
>>> from sentence_transformers.cross_encoder import CrossEncoder
>>> model = CrossEncoder(model_save_path, device="cuda", max_length=64)
>>> sentences = ["今天天气不错", "今天心情不错"]
>>> score = model.predict([sentences])
>>> print(score[0])
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- zh
- Created
- Apr 19, 2022
- Updated
- Apr 20, 2022