In [1]:
from __future__ import print_function
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
%matplotlib inline
In [2]:
from value import ModelScorer
In [5]:
m = ModelScorer("./model/model.cpt.npy.6")
In [42]:
import sys
def draw_pattern_weight(m, stage, corner, direction, state):
idx = 0
weight = 0
for (x, y), corners in m._patterns:
for r, c in corners:
if corner == (r, c) and direction == (x, y):
weight = m._weights[stage][state[0]*3 + state[1] + idx * 9]
idx += 1
corner2 = (corner[0] + direction[0], corner[1] + direction[1])
prt = sys.stdout.write
print("weight: {}".format(weight))
symbols = ['-', '*', 'o']
for r in range(0, 8):
for c in range(0, 8):
if (r, c) == corner:
prt(symbols[state[0]])
elif (r, c) == corner2:
prt(symbols[state[1]])
else:
prt(symbols[0])
prt('\n')
In [0]:
draw_pattern_weight(m, 3, (1, 1), (0, 1), (1, 2))