Downloads · 30 days
16
2% of all-time downloads
cross_encoder_electra-180g-large-discriminator
tuhailong/cross_encoder_electra-180g-large-discriminator
cross_encoder_electra-180g-large-discriminator 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
808
Public
Repo size
2.6 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin1.3 GB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- electra
- Access
- Public
- Created
- Apr 19, 2022
- Updated
- Apr 20, 2022
- SHA
- 94093faa
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-electra-180g-large-discriminator.
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])
Code
train code from https://github.com/TTurn/cross-encoder
- Task
- Text Classification
- Library
- transformers
- Type
- electra
- Languages
- zh
- Created
- Apr 19, 2022
- Updated
- Apr 20, 2022