Commit f9afb912 authored by Jayant Khatkar's avatar Jayant Khatkar

sharing best rollout rather than just the state at the tree node

parent 60855847
......@@ -162,8 +162,8 @@ class Tree:
temp.pop(1, None)
top_n_nodes = sorted(temp, key=temp.get, reverse=True)[:self.comm_n]
X = [self.graph.node[n]["state"] for n in top_n_nodes]
q = [self.graph.node[n]["mu"]**2 for n in top_n_nodes]
X = [self.graph.node[n]["best_rollout"] for n in top_n_nodes if self.graph.node[n]["N"]>0]
q = [self.graph.node[n]["mu"]**2 for n in top_n_nodes if self.graph.node[n]["N"]>0]
self.my_act_dist = ActionDistribution(X,q)
return True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment