본문 바로가기
전공/머신러닝

[ML] 원핫인코딩(One-Hot Encoding)

by mhiiii 2024. 11. 20.
728x90

원-핫 인코딩(One-hot encoding)이란?

원-핫 인코딩은 선택해야하는 선택지의 개수만큼의 차원을 가지면서, 각 선택지의 인덱스에 해당하는 원소에 1, 나머지 원소는 0의 값을 가지게 함

 

원-핫 인코딩으로 표현된 벡터를 원-핫 벡터(one-hot vector)라고 함 

 

보통 다중 클래스 분류 문제에는 각 클래스 간의 관계가 균등하다는 점에서 원-핫 벡터를 많이 사용 

 

다수의 클래스를 분류하는 문제에서는 이진 분류처럼 2개의 숫자 레이블이 아니라 클래스의 개수만큼 숫자 레이블이 필요하기 때문에 비효율적 

 

References

https://wikidocs.net/59678

728x90