Struct MLPRegressor 

pub struct MLPRegressor {
    pub hidden_layer_sizes: Vec<usize>,
    pub activation: Activation,
    pub solver: MLPSolver,
    pub alpha: f32,
    pub learning_rate_init: f32,
    pub max_iter: usize,
    pub tol: f32,
    pub batch_size: usize,
    /* private fields */
}

Multi-Layer Perceptron Regressor

Fields

hidden_layer_sizes: Vec<usize>

activation: Activation

solver: MLPSolver

alpha: f32

learning_rate_init: f32

max_iter: usize

tol: f32

batch_size: usize

Implementations

impl MLPRegressor

pub fn new(hidden_layer_sizes: Vec<usize>) -> Self

pub fn activation(self, act: Activation) -> Self

pub fn max_iter(self, n: usize) -> Self

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

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

pub fn score(&self, x: &Tensor, y: &Tensor) -> f32