siamese-car-matcher

Siamese Car Matcher

Сиамская нейросеть для определения, описывают ли два текста один автомобиль.

Архитектура

• Encoder: bert-base-multilingual-cased (shared)
• Interaction: [A, B, A-B, A*B] → 3072
• Classifier: 3072 → 256 → 64 → 1

Использование

from model import SiameseCarMatcher
from transformers import AutoTokenizer
import torch

model = SiameseCarMatcher("bert-base-multilingual-cased")
model.load_state_dict(torch.load("car_matcher.pt"))
model.eval()