Sentence Similarity
sentence-transformers
PyTorch
Safetensors
Transformers
Korean
bert
text-classification
text-embeddings-inference
Instructions to use leewaay/kpf-bert-base-klueSTS-cross with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use leewaay/kpf-bert-base-klueSTS-cross with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("leewaay/kpf-bert-base-klueSTS-cross") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Transformers
How to use leewaay/kpf-bert-base-klueSTS-cross with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("leewaay/kpf-bert-base-klueSTS-cross") model = AutoModelForSequenceClassification.from_pretrained("leewaay/kpf-bert-base-klueSTS-cross") - Notebooks
- Google Colab
- Kaggle
leewaay/kpf-bert-base-klueSTS-cross
This is a sentence-transformers model: A cross encoder trained with the pretrained jinmang2/kpfbert model on the KLUE STS dataset for sentence similarity tasks. It's specifically designed for direct evaluation of sentence pairs, making it highly effective for Re-Ranking and Augmented SBERT for data labeling tasks aimed at enhancing SBERT.
Usage (Sentence-Transformers)
Using this model becomes easy when you have sentence-transformers installed:
pip install -U sentence-transformers
Then you can use the model like this:
from sentence_transformers import CrossEncoder
pairs = [('Query', 'Paragraph1'), ('Query', 'Paragraph2') , ('Query', 'Paragraph3')]
model = CrossEncoder('leewaay/kpf-bert-base-klueSTS-cross')
scores = model.predict(pairs)
print(scores)
Citing & Authors
- Downloads last month
- 3