Struct LabelPropagation 

pub struct LabelPropagation {
    pub kernel: LPKernel,
    pub gamma: f32,
    pub n_neighbors: usize,
    pub max_iter: usize,
    pub tol: f32,
    /* private fields */
}

Label Propagation for semi-supervised classification

Fields

kernel: LPKernel

gamma: f32

n_neighbors: usize

max_iter: usize

tol: f32

Implementations

impl LabelPropagation

pub fn new() -> Self

pub fn kernel(self, k: LPKernel) -> Self

pub fn gamma(self, g: f32) -> Self

pub fn fit(&mut self, x: &Tensor, y: &Tensor)

pub fn predict(&self, x: &Tensor) -> Tensor

pub fn predict_proba(&self, _x: &Tensor) -> Tensor

Trait Implementations

impl Default for LabelPropagation

fn default() -> Self

Returns the “default value” for a type.