From 1d7d4e14ef3be2cc105773e7510693b74e73e108 Mon Sep 17 00:00:00 2001 From: Tongzheng Ren Date: Thu, 9 Nov 2017 22:40:44 +0800 Subject: [PATCH] fixed some bugs --- utils/text2data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/text2data.py b/utils/text2data.py index 41e0cdf..31d5ade 100644 --- a/utils/text2data.py +++ b/utils/text2data.py @@ -43,9 +43,9 @@ for t in text: new_board = hex2board(line) board = np.concatenate([board, new_board], axis=3) if num % 19 == 16: - if line == '0': + if int(line) == 0: 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') board = np.concatenate([board, new_board], axis=3) boards = np.concatenate([boards, board], axis=0)