Views
No views yet
pip install sentence-transformers1from sentence_transformers import CrossEncoder
2
3model = CrossEncoder("cross-encoder/monoelectra-large", trust_remote_code=True)
4scores = model.predict([
5 ("How many people live in Berlin?", "Berlin had a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers."),
6 ("How many people live in Berlin?", "Berlin is well known for its museums."),
7])
8print(scores)
9# [ 6.016401 -3.6922567]1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4model = AutoModelForSequenceClassification.from_pretrained("cross-encoder/monoelectra-large", trust_remote_code=True)
5tokenizer = AutoTokenizer.from_pretrained("cross-encoder/monoelectra-large")
6
7features = tokenizer(
8 [
9 ("How many people live in Berlin?", "Berlin had a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers."),
10 ("How many people live in Berlin?", "Berlin is well known for its museums."),
11 ],
12 padding=True,
13 truncation=True,
14 return_tensors="pt",
15)
16
17model.eval()
18with torch.no_grad():
19 scores = model(**features).logits.view(-1)
20print(scores)
21# tensor([ 6.0164, -3.6923])