pub struct DenseNNConfig {
pub input_size: usize,
pub hidden_layers: Vec<usize>,
pub activations: Vec<ActivationKind>,
pub output_size: usize,
pub initializer: InitializerKind,
pub loss: LossKind,
pub learning_rate: f64,
pub epochs: usize,
}
Expand description
Configuration for a dense neural network
Fields§
§input_size: usize
§activations: Vec<ActivationKind>
Must have length = hidden_layers.len() + 1
output_size: usize
§initializer: InitializerKind
§loss: LossKind
§learning_rate: f64
§epochs: usize
Auto Trait Implementations§
impl Freeze for DenseNNConfig
impl RefUnwindSafe for DenseNNConfig
impl Send for DenseNNConfig
impl Sync for DenseNNConfig
impl Unpin for DenseNNConfig
impl UnwindSafe for DenseNNConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more