rust-bert icon indicating copy to clipboard operation
rust-bert copied to clipboard

how to use Cross-Encoder for MS Marco by rust-bert?

Open spursy opened this issue 2 years ago • 0 comments

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')])

spursy avatar Sep 26 '23 08:09 spursy