Struct rustlearn::factorization::factorization_machines::FactorizationMachine

pub struct FactorizationMachine {
    // some fields omitted
}

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

Methods

impl FactorizationMachine

fn get_coefficients(&self) -> &Array

fn get_latent_factors(&self) -> &Array

Trait Implementations

impl Decodable for FactorizationMachine

fn decode<__D: Decoder>(__arg_0: &mut __D) -> Result<FactorizationMachine, __D::Error>

impl Encodable for FactorizationMachine

fn encode<__S: Encoder>(&self, __arg_0: &mut __S) -> Result<(), __S::Error>

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<'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>

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>