pub struct CrossEncoder { /* private fields */ }Expand description
Cross-encoder model for relevance scoring
Implementations§
Source§impl CrossEncoder
impl CrossEncoder
Sourcepub fn new(model_name: &str, backend_type: &str) -> RerankingResult<Self>
pub fn new(model_name: &str, backend_type: &str) -> RerankingResult<Self>
Create new cross-encoder with specified backend
Sourcepub fn with_mock_backend() -> Self
pub fn with_mock_backend() -> Self
Create with mock backend for testing
Sourcepub fn score(&self, query: &str, document: &str) -> RerankingResult<f32>
pub fn score(&self, query: &str, document: &str) -> RerankingResult<f32>
Score a single query-document pair
Sourcepub fn batch_score(
&self,
pairs: &[(String, String)],
) -> RerankingResult<Vec<f32>>
pub fn batch_score( &self, pairs: &[(String, String)], ) -> RerankingResult<Vec<f32>>
Score multiple query-document pairs in batch
Sourcepub fn model_name(&self) -> &str
pub fn model_name(&self) -> &str
Get model name
Sourcepub fn set_batch_size(&mut self, batch_size: usize)
pub fn set_batch_size(&mut self, batch_size: usize)
Set batch size
Trait Implementations§
Source§impl Clone for CrossEncoder
impl Clone for CrossEncoder
Source§fn clone(&self) -> CrossEncoder
fn clone(&self) -> CrossEncoder
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreAuto Trait Implementations§
impl Freeze for CrossEncoder
impl !RefUnwindSafe for CrossEncoder
impl Send for CrossEncoder
impl Sync for CrossEncoder
impl Unpin 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> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
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.