Downloads · 30 days
0
irow/dual-bert-IR
dual-bert-IR is a machine learning model from irow. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model uses two BERT models fine-tuned on a contrastive learning objective. One is responsable for short queries, and the other for longer documents that contain the answer to the query. After encoding many docume…
Downloads · 30 days
0
Access
Public
Updated Feb 1, 2023
Repo size
876 MB
Likes
0
Public
Click a slice to open those files.
.pt876 MB · 100%
From the Hugging Face model README
This model uses two BERT models fine-tuned on a contrastive learning objective. One is responsable for short queries, and the other for longer documents that contain the answer to the query. After encoding many documents, one may perform a nearest neighbor search with the query encoding, to fetch the most relevant document.
Take a look at inference.py to see how to perform inference.