hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add role for mcts debug

Browse Source
This commit is contained in:
Dong Yan 2017-12-26 15:07:15 +08:00
parent aa6b5434c6
commit 8f508c790b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
AlphaGo/game.py
View File

@ -73,10 +73,10 @@ class Game:
mcts.search(max_step=100)
if self.debug:
file = open("mcts_debug.log", 'ab')
np.savetxt(file, mcts.root.Q, header="\nQ value : ", fmt='%.4f', newline=", ")
np.savetxt(file, mcts.root.W, header="\nW value : ", fmt='%.4f', newline=", ")
np.savetxt(file, mcts.root.N, header="\nN value : ", fmt="%d", newline=", ")
np.savetxt(file, mcts.root.prior, header="\nprior : ", fmt='%.4f', newline=", ")
np.savetxt(file, mcts.root.Q, header="\n" + self.role + " Q value : ", fmt='%.4f', newline=", ")
np.savetxt(file, mcts.root.W, header="\n" + self.role + " W value : ", fmt='%.4f', newline=", ")
np.savetxt(file, mcts.root.N, header="\n" + self.role + " N value : ", fmt="%d", newline=", ")
np.savetxt(file, mcts.root.prior, header="\n" + self.role + " prior : ", fmt='%.4f', newline=", ")
file.close()
temp = 1
prob = mcts.root.N ** temp / np.sum(mcts.root.N ** temp)