def test_step_funciton():
input_vec = np.array([-1, 0, 1])
output_vec = OutputFunciton.step_function(input_vec)
assert [0, 1, 1] == output_vec.tolist()
input_vec = np.array([-1, 0, 1])
output_vec = OutputFunciton.step_function(input_vec, bottom=-1)
assert [-1, 1, 1] == output_vec.tolist()
def get_updated_state(self, weight_mat, init_state):
# ここのみだとアソシアトロン
updated_state = np.dot(weight_mat, init_state.T)
# TODO : 1 or -1 を返すようにする.
result = []
for d in updated_state:
result.append(OutputFunciton.step_function(d, bottom=-1)[0])
return np.array([result])
