Permalink
Browse files

Fix imports and __init__ issue

  • Loading branch information...
1 parent b7e680b commit 7ecf206644df26ecaee964cb21c77cefb3e8eea9 @brilee committed Oct 26, 2016
Showing with 4 additions and 4 deletions.
  1. +2 −0 main.py
  2. +2 −4 strategies.py
View
@@ -6,6 +6,8 @@
import sys
import gtp as gtp_lib
+from policy import PolicyNetwork
+from features import DEFAULT_FEATURES
from strategies import RandomPlayer, PolicyNetworkBestMovePlayer, MCTS
from load_data_sets import process_raw_data, DataSet
View
@@ -73,10 +73,9 @@ def suggest_move(self, position):
class PolicyNetworkBestMovePlayer(GtpInterface):
def __init__(self, read_file):
- super().__init__()
self.policy_network = PolicyNetwork(DEFAULT_FEATURES.planes, use_cpu=True)
self.read_file = read_file
- self.refresh_network()
+ super().__init__()
def clear(self):
super().clear()
@@ -176,12 +175,11 @@ def select_leaf(self):
class MCTS(GtpInterface):
def __init__(self, read_file, seconds_per_move=5):
- super().__init__()
self.seconds_per_move = seconds_per_move
self.max_rollout_depth = go.N * go.N * 3
self.policy_network = PolicyNetwork(DEFAULT_FEATURES.planes, use_cpu=True)
self.read_file = read_file
- self.refresh_network()
+ super().__init__()
def clear(self):
super().clear()

0 comments on commit 7ecf206

Please sign in to comment.