44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
|
from cup import Cup
|
||
|
|
import numpy as np
|
||
|
|
import tensorflow as tf
|
||
|
|
from network import Network
|
||
|
|
from board import Board
|
||
|
|
import random
|
||
|
|
|
||
|
|
class Restore_bot:
|
||
|
|
|
||
|
|
def __init__(self, sym):
|
||
|
|
self.cup = Cup()
|
||
|
|
self.sym = sym
|
||
|
|
|
||
|
|
self.graph = tf.Graph()
|
||
|
|
with self.graph.as_default():
|
||
|
|
self.session = tf.Session(graph = self.graph)
|
||
|
|
self.network = Network(self.session)
|
||
|
|
self.network.restore_model()
|
||
|
|
|
||
|
|
def roll(self):
|
||
|
|
print("{} rolled: ".format(self.sym))
|
||
|
|
roll = self.cup.roll()
|
||
|
|
print(roll)
|
||
|
|
return roll
|
||
|
|
|
||
|
|
|
||
|
|
def switch(self,cur):
|
||
|
|
return -1 if cur == 1 else 1
|
||
|
|
|
||
|
|
def get_sym(self):
|
||
|
|
return self.sym
|
||
|
|
|
||
|
|
def make_move(self, board, sym, roll):
|
||
|
|
# print(Board.pretty(board))
|
||
|
|
legal_moves = Board.calculate_legal_states(board, sym, roll)
|
||
|
|
legal_list = list(legal_moves)
|
||
|
|
lol = []
|
||
|
|
for move in legal_list:
|
||
|
|
lol.append(self.network.eval_state(np.array(move).reshape(1,26)))
|
||
|
|
print("Found the best state, being:", np.array(lol).argmax())
|
||
|
|
return legal_list[np.array(lol).argmax()]
|
||
|
|
|
||
|
|
# return random.choice(list(legal_moves))
|