SentenceTransformer CrossEncoder

SentenceTransformerCrossEncoder

SentenceTransformerCrossEncodersentence_transformers (opens in a new tab)CrossEncoder (opens in a new tab)와 대응됩니다.

생성자

Aeca와 통신을 위한 Channel 객체와 모델명이 필수적으로 요구됩니다.

SentenceTransformerCrossEncoder(
    channel: Channel,  model_name: str, timeout: int | None = None
)
인자설명타입
channel채널Channel
model_namesentence_transformers (opens in a new tab)에 정의된 모델 이름str
timeout만료 시간int

predict

sentence_pairs간의 유사도를 계산합니다.

SentenceTransformerCrossEncoder.predict(
    sentence_pairs: list[tuple[str, str]]
) -> (
    np.ndarray | None
)
인자설명타입
sentence_pairs문장 튜플의 리스트list[tuple[str, str]]