Struct AttentionSelector 

pub struct AttentionSelector { /* private fields */ }

Implementations

impl AttentionSelector

pub fn new( mechanisms: Vec<Box<dyn DagAttentionMechanism>>, config: SelectorConfig, ) -> Self

pub fn select(&self) -> usize

Select mechanism using UCB1 algorithm

pub fn update(&mut self, mechanism_idx: usize, reward: f32)

Update rewards after execution

pub fn get_mechanism(&self, idx: usize) -> Option<&dyn DagAttentionMechanism>

Get the selected mechanism

pub fn get_mechanism_mut( &mut self, idx: usize, ) -> Option<&mut Box<dyn DagAttentionMechanism>>

Get mutable reference to mechanism for updates

pub fn stats(&self) -> HashMap<&'static str, MechanismStats>

Get statistics for all mechanisms

pub fn best_mechanism(&self) -> Option<usize>

Get the best performing mechanism based on average reward

pub fn reset(&mut self)

Reset all statistics

pub fn forward( &mut self, dag: &QueryDag, ) -> Result<(AttentionScores, usize), AttentionError>

Forward pass using selected mechanism