Struct ducky_learn::layers::Dense1d

pub struct Dense1d { /* private fields */ }

Implementations

impl Dense1d

pub fn from(
    activation: fn(_: Array1<f64>) -> Array1<f64>,
    weights: Array2<f64>,
    bias: Array1<f64>
) -> Self

Create Dense1d layer with full control over every part of the layer

Arguments

• activation: Activation function of whole 1d array
• weights: 2d array that has to be of shape( output, input )
• bias: 1d array of basis that has to be the size of the output

returns: Dense1d

Examples

use ducky_learn::layers::*;
use ndarray::{arr1, arr2};

let layer = Dense1d::from(
                |x| x, // Activation function that is does nothing
                arr2(&[[1., 1.], [1., 1.]]), // 2x2 array
                arr1(&[1., 1.]) // len 2
            );

Trait Implementations

impl FeedForward1d for Dense1d

fn pass(&self, input_array: Array1<f64>) -> Array1<f64>

Feeds forward the 1d array through the layer.