Downloads · 30 days
9
0% of all-time downloads
PitKoro/cross-encoder-ru-msmarco-passage
cross-encoder-ru-msmarco-passage is a machine learning model from PitKoro. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for sentence-transformers. The card lists the license as apache-2.0.
Downloads · 30 days
9
0% of all-time downloads
All-time downloads
9.5K
Public
Parameters
178M
1.4 GB on disk
Likes
3
Public
Click a slice to open those files.
.bin711 MB · 50%
From the Hugging Face model README
from transformers import AutoModel, AutoTokenizer, AutoModelForSequenceClassification
import torch
sigmoid_fn = torch.nn.Sigmoid()
# model = AutoModelForSequenceClassification.from_pretrained("/home/jovyan/pakorolev/ranker/deep_pavlov_mrr_0_8413")
# tokenizer = AutoTokenizer.from_pretrained("/home/jovyan/pakorolev/ranker/deep_pavlov_mrr_0_8413")
model = AutoModelForSequenceClassification.from_pretrained("PitKoro/cross-encoder-ru-msmarco-passage")
tokenizer = AutoTokenizer.from_pretrained("PitKoro/cross-encoder-ru-msmarco-passage")
text = [['привет', 'привет'],['привет', 'пока']]
tokenized = tokenizer(text, return_tensors='pt')
logits = model(**tokenized).logits
output = sigmoid_fn(logits.flatten())
print(output)
from sentence_transformers.cross_encoder import CrossEncoder
model = CrossEncoder("PitKoro/cross-encoder-ru-msmarco-passage", max_length=512)
text = [['привет', 'привет'],['привет', 'пока']]
output = model.predict(text)
print(output)