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))