تبدیل آرایه numpy به one hot

Question

آیا numpy تابعی داره که آرایه ای را دریافت و onehot آن آرایه را بر گردونه؟

Answer 1

خودم نوشتم:

def getOnehot(list,max_count):
    len_list = len(list)
    result = np.zeros((len_list, max_count))
    result[np.arange(len_list), list] = 1
    return  result

 

Answer 2

با کراس هم می تونید به این صورت:

from keras.utils import np_utils
y_train = np_utils.to_categorical(y_train)