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

fixed some bugs

Browse Source
This commit is contained in:
Tongzheng Ren 2017-11-09 22:40:44 +08:00
parent e0b59a3404
commit 1d7d4e14ef
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/text2data.py
View File

@ -43,9 +43,9 @@ for t in text:
new_board = hex2board(line) new_board = hex2board(line)
board = np.concatenate([board, new_board], axis=3) board = np.concatenate([board, new_board], axis=3)
if num % 19 == 16: if num % 19 == 16:
if line == '0': if int(line) == 0:
new_board = np.ones([1, 19 ,19 ,1], dtype='int8') new_board = np.ones([1, 19 ,19 ,1], dtype='int8')
if line == '1': if int(line) == 1:
new_board = np.zeros([1, 19, 19, 1], dtype='int8') new_board = np.zeros([1, 19, 19, 1], dtype='int8')
board = np.concatenate([board, new_board], axis=3) board = np.concatenate([board, new_board], axis=3)
boards = np.concatenate([boards, board], axis=0) boards = np.concatenate([boards, board], axis=0)