pub struct CrossEncoder { /* private fields */ }Expand description
Stateful cross-encoder that tracks the total number of pairs scored.
Implementations§
Source§impl CrossEncoder
impl CrossEncoder
Sourcepub fn new(config: CrossEncoderConfig) -> Self
pub fn new(config: CrossEncoderConfig) -> Self
Create a new CrossEncoder with the given configuration.
Sourcepub fn score(&mut self, pair: &CandidatePair) -> f32
pub fn score(&mut self, pair: &CandidatePair) -> f32
Score a single (query, document) pair using token-overlap similarity.
Sourcepub fn score_batch(&mut self, pairs: &[CandidatePair]) -> Vec<f32>
pub fn score_batch(&mut self, pairs: &[CandidatePair]) -> Vec<f32>
Score multiple pairs at once. Respects batch_size but processes
sequentially in the simulation (no concurrency overhead).
Sourcepub fn rerank(
&mut self,
query: &str,
candidates: &[String],
initial_scores: &[f32],
) -> Vec<RerankResult>
pub fn rerank( &mut self, query: &str, candidates: &[String], initial_scores: &[f32], ) -> Vec<RerankResult>
Re-rank candidates against query.
Returns a list of RerankResult sorted by rerank_score descending,
with rank fields assigned from 1.
Sourcepub fn top_k(
&mut self,
query: &str,
candidates: &[String],
initial_scores: &[f32],
k: usize,
) -> Vec<RerankResult>
pub fn top_k( &mut self, query: &str, candidates: &[String], initial_scores: &[f32], k: usize, ) -> Vec<RerankResult>
Like rerank but truncates the output to the top-k results.
Sourcepub fn total_scored(&self) -> u64
pub fn total_scored(&self) -> u64
Total number of individual pairs that have been scored so far.
Auto Trait Implementations§
impl Freeze for CrossEncoder
impl RefUnwindSafe for CrossEncoder
impl Send for CrossEncoder
impl Sync for CrossEncoder
impl Unpin for CrossEncoder
impl UnsafeUnpin for CrossEncoder
impl UnwindSafe for CrossEncoder
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<T> PolicyExt for Twhere
T: ?Sized,
impl<T> PolicyExt for Twhere
T: ?Sized,
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.