Struct TransitionModel 

pub struct TransitionModel { /* private fields */ }

Model for predicting state transitions

Implementations

impl TransitionModel

pub fn new(bins: usize) -> Self

pub fn record( &mut self, state: &Observation, action: &Action, next_state: &Observation, )

Record a transition

pub fn get_transition_probs( &self, state: &Observation, action: &Action, ) -> HashMap<StateKey, f64>

Get probability distribution of next states

pub fn predict(&self, state: &Observation, action: &Action) -> Option<StateKey>

Get most likely next state

pub fn get_count(&self, state: &Observation, action: &Action) -> u64

Get transition count