Tensorflow - to_categorical함수로 원 핫 인코딩으로 변경하기

 

to_categorical

 

 

 

분류의 문제일때 y_train 값이 레이블 인코딩일 경우

 

카테고리컬 데이터를 원 핫 인코딩으로 변경해야 하는데,

 

Tensorflow 에서 쉽게 원 핫 인코딩으로 변경 할 수 있게끔 라이브러리를 제공한다.

 

이때 사용하는 라이브러리 함수가 to_categorical() 함수이다.

 

 

# num_classes == output의 노드 갯수와 같아야 한다.

y_train = to_categorical(y_train, num_classes=10 )

y_train[0]

 

 

# y_train을 바꿔줬으면 y_test도 바꿔준다.

y_test = to_categorical(y_test, 10)

y_test