Pownie/backgammon
1
0
Fork 0
Code Issues 6 Pull Requests Packages Projects Releases Wiki Activity

Added comments for Christoffer!

Browse Source
This commit is contained in:
Alexander Munch-Hansen 2018-05-10 19:25:28 +02:00
parent 1aedc23de1
commit 6131d5b5f4
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
network.py
View File

@ -307,6 +307,7 @@ class Network:
list_of_moves = []
# Prepping of data
for idx, board in enumerate(boards):
all_board_moves = []
for roll in all_rolls:
@ -317,14 +318,16 @@ class Network:
list_of_moves.append(np.array(all_board_moves))
print(time.time() - start)
start = time.time()
# Running data through networks
all_scores = [self.model.predict_on_batch(board) for board in list_of_moves]
transformed_scores = [x if player == 1 else (1-x) for x in all_scores]
scores_means = [tf.reduce_mean(score) for score in all_scores]
transformed_means = [tf.reduce_mean(score) for score in transformed_scores]
print(time.time() - start)
return ([scores_means, transformed_means])