Skip to main content

create_network

Function create_network 

Source
pub fn create_network(
    layer_sizes: &[usize],
    activation: ActivationType,
) -> NeuralNet
Expand description

Create a fully-connected network with the given layer sizes and random weights.

layer_sizes must contain at least 2 entries (input + output). All hidden layers use activation; the output layer uses Linear.