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

Merge branch 'master' of https://github.com/sproblvem/tianshou

Browse Source
This commit is contained in:
rtz19970824 2017-12-22 13:47:39 +08:00
commit d42a76f8f3
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
AlphaGo/model.py
View File

@ -215,6 +215,7 @@ class ResNet(object):
self.training_data['states'].append(states) self.training_data['states'].append(states)
self.training_data['probs'].append(probs) self.training_data['probs'].append(probs)
self.training_data['winner'].append(winner) self.training_data['winner'].append(winner)
if len(self.training_data['states']) == self.window_length:
training_data['states'] = np.concatenate(self.training_data['states'], axis=0) training_data['states'] = np.concatenate(self.training_data['states'], axis=0)
training_data['probs'] = np.concatenate(self.training_data['probs'], axis=0) training_data['probs'] = np.concatenate(self.training_data['probs'], axis=0)
training_data['winner'] = np.concatenate(self.training_data['winner'], axis=0) training_data['winner'] = np.concatenate(self.training_data['winner'], axis=0)