entrenar 0.7.12

Training & Optimization library with autograd, LoRA, quantization, and model merging
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79

//! Provenance node types for incident reconstruction.

use serde::{Deserialize, Serialize};

/// Unique identifier for a node in the provenance graph
pub type NodeId = u64;

/// Provenance graph node types
#[derive(Clone, Debug, Serialize, Deserialize)]
pub enum ProvenanceNode {
    /// Raw sensor/input data
    Input {
        /// Source identifier
        source: String,
        /// Timestamp in nanoseconds
        timestamp_ns: u64,
        /// Hash of the input data
        hash: u64,
    },

    /// Preprocessing transformation
    Transform {
        /// Operation name
        operation: String,
        /// References to input nodes
        input_refs: Vec<NodeId>,
    },

    /// Model inference
    Inference {
        /// Model identifier
        model_id: String,
        /// Model version
        model_version: String,
        /// Confidence of the inference
        confidence: f32,
        /// Output value
        output: f32,
    },

    /// Post-processing or sensor fusion
    Fusion {
        /// Fusion strategy
        strategy: String,
        /// References to input nodes
        input_refs: Vec<NodeId>,
    },

    /// Final action/output
    Action {
        /// Action type
        action_type: String,
        /// Confidence
        confidence: f32,
        /// Alternatives considered
        alternatives: Vec<(String, f32)>,
    },
}

impl ProvenanceNode {
    /// Get the timestamp if available
    pub fn timestamp_ns(&self) -> Option<u64> {
        match self {
            ProvenanceNode::Input { timestamp_ns, .. } => Some(*timestamp_ns),
            _ => None,
        }
    }

    /// Get node type as string
    pub fn type_name(&self) -> &'static str {
        match self {
            ProvenanceNode::Input { .. } => "Input",
            ProvenanceNode::Transform { .. } => "Transform",
            ProvenanceNode::Inference { .. } => "Inference",
            ProvenanceNode::Fusion { .. } => "Fusion",
            ProvenanceNode::Action { .. } => "Action",
        }
    }
}