ton/LLMMCTS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
narawat lamaiin
2024-06-01 08:17:48 +07:00
parent 74a4190440
commit 2378ddfa70
4 changed files with 458 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/interface.jl
View File

@@ -47,7 +47,7 @@ julia>
# Signature
"""
function runMCTS(
workDict::T1,
config::T1,
initialState,
decisionMaker::Function,
evaluator::Function,
@@ -74,11 +74,11 @@ function runMCTS(
# do nothing then go directly to backpropagation
backpropagate(leafNode, node.reward)
else
expand(workDict, node, decisionMaker, evaluator, reflector, transition;
expand(config, node, decisionMaker, evaluator, reflector, transition;
totalsample=totalsample)
leafNode = selectChildNode(node)
simTrajectoryReward, terminalstate = simulate(workDict, leafNode, decisionMaker, evaluator,
reflector; maxDepth=maxDepth, totalsample=totalsample)
simTrajectoryReward, terminalstate = simulate(config, leafNode, decisionMaker, evaluator,
reflector, transition; maxDepth=maxDepth, totalsample=totalsample)
if terminalstate !== nothing #XXX not sure why I need this
terminalstate[:totalTrajectoryReward] = simTrajectoryReward
end