rust-bert
rust-bert copied to clipboard
how to use Cross-Encoder for MS Marco by rust-bert?
The Cross-Encoder for MS Marco hugging face address is here
I want to use predict func like the followed python code:
from sentence_transformers import CrossEncoder
model = CrossEncoder('model_name', max_length=512)
scores = model.predict([('Query', 'Paragraph1'), ('Query', 'Paragraph2') , ('Query', 'Paragraph3')])