ast_toolbox.mcts.AST_MCTS module¶
-
ast_toolbox.mcts.AST_MCTS.
explore_getAction
(ast)[source]¶ Get the exploration function from ast.
Parameters: ast ( ast_toolbox.mcts.AdaptiveStressTest.AdaptiveStressTesting
) – The AST object.
-
ast_toolbox.mcts.AST_MCTS.
rollout_getAction
(ast)[source]¶ Get the rollout function from ast.
Parameters: ast ( ast_toolbox.mcts.AdaptiveStressTest.AdaptiveStressTesting
) – The AST object.
-
ast_toolbox.mcts.AST_MCTS.
stress_test
(ast, mcts_params, top_paths, verbose=True, return_tree=False)[source]¶ Run stress test with mode 1 (search with single tree).
Parameters: - ast (
ast_toolbox.mcts.AdaptiveStressTest.AdaptiveStressTesting
) – The AST object. - mcts_params (
ast_toolbox.mcts.MCTSdpw.DPWParams
) – The mcts parameters. - top_paths (
ast_toolbox.mcts.BoundedPriorityQueues
) – The bounded priority queue to store top-rewarded trajectories. - verbose (bool, optional) – Whether to logging test information
- return_tree (bool, optional) – Whether to return the search tree
Returns: - results (
ast_toolbox.mcts.AdaptiveStressTest.AdaptiveStressTesting
) – The bounded priority queue storing top-rewarded trajectories. - tree (dict) – The resulting searching tree.
- ast (
-
ast_toolbox.mcts.AST_MCTS.
stress_test2
(ast, mcts_params, top_paths, verbose=True, return_tree=False)[source]¶ Run stress test with mode 2 (search with multiple trees).
Parameters: - ast (
ast_toolbox.mcts.AdaptiveStressTest.AdaptiveStressTesting
) – The AST object. - mcts_params (
ast_toolbox.mcts.MCTSdpw.DPWParams
) – The mcts parameters. - top_paths (
ast_toolbox.mcts.BoundedPriorityQueues
) – The bounded priority queue to store top-rewarded trajectories. - verbose (bool, optional) – Whether to logging test information
- return_tree (bool, optional) – Whether to return the search tree
Returns: - results (
ast_toolbox.mcts.AdaptiveStressTest.AdaptiveStressTesting
) – The bounded priority queue storing top-rewarded trajectories. - tree (dict) – The resulting searching tree.
- ast (