SentenceTransformerCrossEncoder
SentenceTransformerCrossEncoder
는 sentence_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_name | sentence_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]] |