In the code seem to be different from the setting of the paper.

[roseif]

rand_h_1 = np.random.randint(output_dim, size=self.input_dim1) and rand_s_1 = 2 * np.random.randint(2, size=self.input_dim1) - 1 in the code seem to be different from the setting of the paper. In the paper, hk is {1,2,...,k}, sk is {+1 ,-1}.