sarek 0.1.0

A work-in-progress, experimental neural network library utilizing TensorFlow Keras
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

use {
    crate::{
        core::{
            name::{
                Name
            },
            shape::{
                Shape
            }
        },
        nn::{
            layers::{
                IntoLayerIter,
                Layer,
                LayerPrototype
            }
        }
    }
};

/// A neural network model.
#[derive(Clone, Debug)]
pub struct Model {
    pub(crate) layers: Vec< Layer >,
    input_shape: Shape
}

impl Model {
    pub fn new_sequential< S, I >( input_shape: S, layers: I ) -> Model
        where S: Into< Shape >,
              I: IntoLayerIter
    {
        Model {
            layers: layers.into_layer_iter().collect(),
            input_shape: input_shape.into()
        }
    }

    pub fn input_shape( &self ) -> Shape {
        self.input_shape.clone()
    }

    pub(crate) fn get_layer_and_input_shape( &self, layer_name: &Name ) -> Option< (&Layer, Shape) > {
        let mut input_shape = self.input_shape();
        let mut target_layer = None;
        for layer in &self.layers {
            if *layer.name() == *layer_name {
                target_layer = Some( layer );
                break;
            }

            input_shape = layer.output_shape( &input_shape );
        }

        target_layer.map( |layer| (layer, input_shape) )
    }

    pub(crate) fn output_shape( &self ) -> Shape {
        let mut input_shape = self.input_shape();
        for layer in &self.layers {
            input_shape = layer.output_shape( &input_shape );
        }
        input_shape
    }
}