def one_hot(hypno, n_categories): enc = OneHotEncoder(n_values=n_categories) hypno = enc.fit_transform(hypno).toarray() return np.array(hypno,'int32')