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