dag-ml-core 0.2.1

Core graph, phase, OOF and deterministic control contracts for dag-ml.
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

use serde::{Deserialize, Serialize};

#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum Phase {
    Compile,
    Plan,
    FitCv,
    Select,
    Refit,
    Predict,
    Explain,
}

impl Phase {
    pub fn is_training(self) -> bool {
        matches!(self, Self::FitCv | Self::Refit)
    }

    /// Stable SCREAMING_SNAKE_CASE label matching the serde representation. Used
    /// as the `phase` field in ADR-12 observability spans.
    pub fn as_str(self) -> &'static str {
        match self {
            Self::Compile => "COMPILE",
            Self::Plan => "PLAN",
            Self::FitCv => "FIT_CV",
            Self::Select => "SELECT",
            Self::Refit => "REFIT",
            Self::Predict => "PREDICT",
            Self::Explain => "EXPLAIN",
        }
    }
}