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_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.