Struct FactorizationMachine

pub struct FactorizationMachine { /* private fields */ }

A two-class factorization machine implemented using stochastic gradient descent.

Implementations

impl FactorizationMachine

pub fn get_coefficients(&self) -> &Array

pub fn get_latent_factors(&self) -> &Array

Trait Implementations

impl Clone for FactorizationMachine

fn clone(&self) -> FactorizationMachine

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<'de> Deserialize<'de> for FactorizationMachine

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl<'a, T> ParallelSupervisedModel<&'a T> for FactorizationMachine

where &'a T: RowIterable, T: IndexableMatrix + Sync,

fn fit_parallel( &mut self, X: &'a T, y: &Array, num_threads: usize, ) -> Result<(), &'static str>

impl Serialize for FactorizationMachine

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl<'a, T> SupervisedModel<&'a T> for FactorizationMachine

where &'a T: RowIterable, T: IndexableMatrix,

fn fit(&mut self, X: &'a T, y: &Array) -> Result<(), &'static str>

fn decision_function(&self, X: &'a T) -> Result<Array, &'static str>

fn predict(&self, x: T) -> Result<Array, &'static str>