개발/머신러닝

파이썬 머신러닝 - 레이블 y 값을 원핫인코딩으로 바꾸기.

웅'jk 2022. 12. 29. 17:02

딥러닝에서 y의 값이 레이블 형식인데 원 핫 으로 사용하고 싶다면

tf.keras.utils.to_categorical(y_train,num_classes=10 )

num_classes 는 y 데이터의 분류할 갯수이다. 즉 지금은 10개로 분류하겠다는 뜻이다.

 

이렇게 y를 원 핫 인코딩 으로 변경해서 사용할 경우 compile 에서 loss 함수가 달라지 게 된다.

model.compile(optimizer='adam',loss = 'categorical_crossentropy',metrics=['accuracy'] )

이런식으로 카테고리컬 크로스엔트로피로 사용해야한다.