nnrs 0.2.3

Rust Neural Network Library
Documentation 
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

use serde::{Deserialize, Serialize};
use std::cmp::Ordering;

/// Assigned to a node to indicate which layer it is in.
#[derive(Eq, Debug, Clone, Copy, Serialize, Deserialize)]
pub enum LayerID {
    /// The input layer.
    InputLayer,

    /// A hidden layer, with an id.
    HiddenLayer(usize),

    /// The output layer.
    OutputLayer,
}

impl LayerID {
    /// Returns the next layer id.
    pub fn is_hidden(&self) -> bool {
        matches!(self, LayerID::HiddenLayer(_))
    }
}

impl Ord for LayerID {
    fn cmp(&self, other: &Self) -> Ordering {
        match (self, other) {
            (LayerID::InputLayer, LayerID::InputLayer) => Ordering::Equal,
            (LayerID::InputLayer, _) => Ordering::Less,
            (LayerID::OutputLayer, LayerID::OutputLayer) => Ordering::Equal,
            (LayerID::OutputLayer, _) => Ordering::Greater,
            (LayerID::HiddenLayer(x), LayerID::HiddenLayer(y)) => x.cmp(y),
            (_, LayerID::InputLayer) => Ordering::Greater,
            (_, LayerID::OutputLayer) => Ordering::Less,
        }
    }
}

impl PartialEq for LayerID {
    fn eq(&self, other: &Self) -> bool {
        match (self, other) {
            (LayerID::InputLayer, LayerID::InputLayer) => true,
            (LayerID::OutputLayer, LayerID::OutputLayer) => true,
            (LayerID::HiddenLayer(x), LayerID::HiddenLayer(y)) => x == y,
            _ => false,
        }
    }
}

impl PartialOrd for LayerID {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}