Struct KstemitModel 

pub struct KstemitModel {
    pub config: KstemitConfig,
    pub temporal: TemporalGlu,
    pub spatial: GraphSageLayer,
    pub physics_proj: Linear,
    pub alpha_logit: f32,
    pub head: Linear,
}

K-STEMIT-style spatio-temporal GNN for sonobuoy arrays.

temporal window ──► TemporalGlu ──► h_t ──┐
                                          │ α
node feats + AGG ─► GraphSAGE ───► h_s ───┼──► fuse ──► Linear head ──► logits
                                          │
physics priors ───► Linear ──────► h_p ───┘ (γ residual)

Fields

config: KstemitConfig

temporal: TemporalGlu

spatial: GraphSageLayer

physics_proj: Linear

alpha_logit: f32

Unconstrained logit; α = sigmoid(alpha_logit).

head: Linear

Implementations

impl KstemitModel

pub fn new(config: KstemitConfig) -> Self

Construct a freshly initialized model.

pub fn alpha(&self) -> f32

Current fusion weight α ∈ (0, 1).

pub fn forward_node( &self, graph: &SensorGraph, node_idx: usize, ) -> ForwardOutput

Forward for one node index on a SensorGraph.

pub fn inputs_for( &self, graph: &SensorGraph, node_idx: usize, ) -> (Vec<f32>, Vec<f32>, Vec<f32>, Vec<f32>)

Build model inputs for a node (public for training caches).

pub fn forward_tensors( &self, x_self: &[f32], x_agg: &[f32], series: &[f32], physics: &[f32], ) -> ForwardOutput

Forward from pre-built tensors.

pub fn predict(&self, graph: &SensorGraph, node_idx: usize) -> usize

Predicted class (argmax).

pub fn loss_node( &self, graph: &SensorGraph, node_idx: usize, label: usize, ) -> f32

Loss for one labelled node.

pub fn train_step( &mut self, graph: &SensorGraph, node_idx: usize, label: usize, lr: f32, weight_decay: f32, ) -> f32

One SGD step on a single (graph, node, label) sample.

Analytical gradients through GLU, GraphSAGE, fusion α, and head.

Trait Implementations

impl Clone for KstemitModel

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for KstemitModel

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.