Struct ClassifierConfig

pub struct ClassifierConfig { /* private fields */ }

A config for a Classifier.

Implementations

impl ClassifierConfig

pub fn new() -> Self

Create a new config.

pub fn layers_dims(self, layers_dims: impl IntoIterator<Item = usize>) -> Self

Set the dimensions of the layers.

pub fn dropout_rate(self, dropout_rate: f32) -> Self

Set the dropout rate.

pub fn classes(self, classes: u32) -> Self

Set the number of classes. This is required if Class::CLASSES is not defined for the type you are classifying.

Trait Implementations

impl Clone for ClassifierConfig

fn clone(&self) -> ClassifierConfig

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ClassifierConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ClassifierConfig

fn default() -> Self

Returns the “default value” for a type.