Struct MLPClassifier 

pub struct MLPClassifier {
    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,
    pub momentum: f32,
    /* private fields */
}

Multi-Layer Perceptron Classifier

Fields

hidden_layer_sizes: Vec<usize>

activation: Activation

solver: MLPSolver

alpha: f32

learning_rate_init: f32

max_iter: usize

tol: f32

batch_size: usize

momentum: f32

Implementations

impl MLPClassifier

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

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

pub fn solver(self, solver: MLPSolver) -> Self

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

pub fn learning_rate(self, lr: 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

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