Struct RBFNetwork 

pub struct RBFNetwork {
    pub n_centers: usize,
    pub gamma: f32,
    pub max_iter: usize,
    pub tol: f32,
    /* private fields */
}

RBF Network for classification and regression

Fields

n_centers: usize

gamma: f32

max_iter: usize

tol: f32

Implementations

impl RBFNetwork

pub fn new(n_centers: usize) -> 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