pub struct Trainer { /* private fields */ }Expand description
Joint model trainer using softmax-margin objective.
Implementations§
Source§impl Trainer
impl Trainer
Sourcepub fn new(config: TrainingConfig) -> Self
pub fn new(config: TrainingConfig) -> Self
Create a new trainer.
Sourcepub fn add_example(&mut self, example: TrainingExample)
pub fn add_example(&mut self, example: TrainingExample)
Add a training example.
Sourcepub fn add_examples(
&mut self,
examples: impl IntoIterator<Item = TrainingExample>,
)
pub fn add_examples( &mut self, examples: impl IntoIterator<Item = TrainingExample>, )
Add multiple training examples.
Sourcepub fn get_weights(&self) -> &JointWeights
pub fn get_weights(&self) -> &JointWeights
Get current weights.
Sourcepub fn get_loss_history(&self) -> &[f64]
pub fn get_loss_history(&self) -> &[f64]
Get loss history.
Auto Trait Implementations§
impl Freeze for Trainer
impl RefUnwindSafe for Trainer
impl Send for Trainer
impl Sync for Trainer
impl Unpin for Trainer
impl UnsafeUnpin for Trainer
impl UnwindSafe for Trainer
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 more