burn_dragon_core 0.5.0

burn dragon core model and utilities
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

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum StructuredStepMode {
    Observe,
    Refine,
    Predict,
}

impl StructuredStepMode {
    pub const COUNT: usize = 3;

    pub const fn index(self) -> usize {
        match self {
            Self::Observe => 0,
            Self::Refine => 1,
            Self::Predict => 2,
        }
    }

    pub const fn as_str(self) -> &'static str {
        match self {
            Self::Observe => "observe",
            Self::Refine => "refine",
            Self::Predict => "predict",
        }
    }

    pub const fn temporal_dt(self) -> usize {
        match self {
            Self::Observe => 0,
            Self::Refine => 0,
            Self::Predict => 1,
        }
    }

    pub const fn resets_prediction_age(self) -> bool {
        matches!(self, Self::Observe)
    }
}