Struct Params

pub struct Params { /* private fields */ }

A set of parameters.

Implementations

impl Params

pub fn new() -> Self

Returns a new set of parameters.

pub fn learning_rate(&mut self, value: f32) -> &mut Self

Sets the learning rate.

pub fn lambda(&mut self, value: f32) -> &mut Self

Sets the regularization parameter.

pub fn iterations(&mut self, value: i32) -> &mut Self

Sets the number of iterations.

pub fn factors(&mut self, value: i32) -> &mut Self

Sets the number of latent factors.

pub fn normalization(&mut self, value: bool) -> &mut Self

Sets whether to use instance-wise normalization.

pub fn auto_stop(&mut self, value: bool) -> &mut Self

Sets whether to stop at the iteration that achieves the best validation loss.

pub fn quiet(&mut self, value: bool) -> &mut Self

Sets whether to use quiet mode (no output).

pub fn on_disk(&mut self, value: bool) -> &mut Self

Sets whether to use on-disk training.

pub fn train<P: AsRef<Path>>(&self, tr_path: P) -> Result<Model, Error>

Trains a model.

pub fn train_eval<P: AsRef<Path>, Q: AsRef<Path>>( &self, tr_path: P, va_path: Q, ) -> Result<Model, Error>

Trains a model and performs cross-validation.

Trait Implementations

impl Debug for Params

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

Formats the value using the given formatter.

impl Default for Params

fn default() -> Self

Returns the “default value” for a type.