더북(TheBook)

코드 4-14 레이블 인코딩하기

def to_one_hot(labels, dimension=46):
    results = np.zeros((len(labels), dimension))
    for i, label in enumerate(labels):
        results[i, label] = 1.
    return results
y_train = to_one_hot(train_labels)    
y_test = to_one_hot(test_labels) 

훈련 레이블 벡터 변환

테스트 레이블 벡터 변환

케라스에는 이를 위한 내장 함수가 있습니다.

from keras.utils.np_utils import to_categorical

y_train = to_categorical(train_labels)
y_test = to_categorical(test_labels)
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.