Struct ModelState 

pub struct ModelState<T, State> {
    pub inner: T,
    /* private fields */
}

State-based wrapper for ML models

Fields

inner: T

Implementations

impl<T> ModelState<T, Untrained>

pub fn new(inner: T) -> Self

Create a new untrained model

pub fn train(self) -> ModelState<T, Trained>

Transition to trained state (only available for untrained models)

impl<T> ModelState<T, Trained>

pub fn predict<F, Input, Output>(&self, predict_fn: F, input: Input) -> Output

where F: Fn(&T, Input) -> Output,

Predict (only available for trained models)

pub fn reset(self) -> ModelState<T, Untrained>

Reset to untrained state

Trait Implementations

impl<T: Clone, State: Clone> Clone for ModelState<T, State>

fn clone(&self) -> ModelState<T, State>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<T: Debug, State: Debug> Debug for ModelState<T, State>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.