Struct MCTS

pub struct MCTS<D: Domain> { /* private fields */ }

The state of a running planner instance.

Implementations

impl<D: Domain> MCTS<D>

pub fn new( initial_state: D::State, root_agent: AgentId, config: MCTSConfiguration, ) -> Self

Instantiates a new search tree for the given state, with idle tasks for all agents and starting at tick 0.

pub fn new_with_tasks( initial_state: D::State, root_agent: AgentId, start_tick: u64, tasks: ActiveTasks<D>, config: MCTSConfiguration, state_value_estimator: Box<dyn StateValueEstimator<D> + Send>, early_stop_condition: Option<Box<EarlyStopCondition>>, ) -> Self

Instantiates a new search tree for the given state, with active tasks for all agents and starting at a given tick.

pub fn best_task_at_root(&mut self) -> Option<Box<dyn Task<D>>>

Returns the best task, using exploration value of 0.

pub fn best_task_with_history( &self, task_history: &HashMap<AgentId, ActiveTask<D>>, ) -> Box<dyn Task<D>>

where D: DomainWithPlanningTask,

Returns the best task, following a given recent task history, in case planning tasks are used.

pub fn q_value_at_root(&self, agent: AgentId) -> f32

Returns the q-value at root.

pub fn run(&mut self) -> Option<Box<dyn Task<D>>>

Executes the MCTS search.

Returns the current best task, if there is at least one task for the root node.

pub fn initial_state(&self) -> &D::State

Returns the initial state at the root of the planning tree.

pub fn start_tick(&self) -> u64

Returns the tick at the root of the planning tree.

pub fn agent(&self) -> AgentId

Returns the agent the tree searches for.

pub fn min_max_range(&self, agent: AgentId) -> Range<AgentValue>

Returns the range of minimum and maximum global values.

pub fn nodes(&self) -> impl Iterator<Item = (&Node<D>, &Edges<D>)>

Returns an iterator over all nodes and edges in the tree.

pub fn root_node(&self) -> Node<D>

Returns the root node of the search tree.

pub fn get_edges(&self, node: &Node<D>) -> Option<&Edges<D>>

Returns the edges associated to a given node.

pub fn seed(&self) -> u64

Returns the seed of the tree.

pub fn node_count(&self) -> usize

Returns the number of nodes.

pub fn edge_count(&self) -> usize

Returns the number of nodes.

pub fn time(&self) -> Duration

Returns the duration of the last run.

pub fn size(&self, task_size: fn(&dyn Task<D>) -> usize) -> usize

Returns an estimation of the memory footprint of the MCTS struct.