ms-marco-MiniLM-L-6-v2 (Cross-Encoder) โ€” ONNX

ONNX export of ms-marco-MiniLM-L-6-v2, a cross-encoder model trained on MS MARCO passage ranking data. Scores query-document pairs for search result reranking.

Mirrored for use with inference4j, an inference-only AI library for Java.

Original Source

Usage with inference4j

try (MiniLMReranker reranker = MiniLMReranker.fromPretrained("models/ms-marco-MiniLM-L-6-v2")) {
    float score = reranker.score("What is Java?", "Java is a programming language.");
    System.out.printf("Relevance score: %.4f%n", score);
}

Model Details

Property Value
Architecture MiniLM-L6 cross-encoder (6 layers, 384 hidden)
Task Query-document relevance scoring / search reranking
Training data MS MARCO passage ranking
Max sequence length 512
Output Single relevance score per query-document pair
Original framework PyTorch (sentence-transformers cross-encoder)
ONNX export By Xenova (Transformers.js)

License

This model is licensed under the Apache License 2.0. Original model by sentence-transformers, ONNX export by Xenova.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support