[파이토치] 원-핫 인코딩 하는 법 (one-hot encoding)
원-핫 이란? 0과 1로만 이루어진 어떤 집합이 있을 때 하나의 원소만 1이고 나머지는 0인 경우를 원-핫 이라고 합니다. 원-핫 인코딩이란? 어떤 데이터를 원-핫 데이터로 만드는 것을 원-핫 인코딩이라고 부릅니다. 0,1,2,3,4 라는 다섯개의 레이블로 분류된 어떤 데이터가 있다고 합시다. 이때 레이블을 원-핫으로 바꾸는 것을 원-핫 인코딩이라고 합니다. 0,1,2,3,4 를 아래와 같이 바꾸는 것입니다. 0 -> [1,0,0,0,0] 1 -> [0,1,0,0,0] 2 -> [0,0,1,0,0] 3 -> [0,0,0,1,0] 4 -> [0,0,0,0,1] 파이토치에서 원-핫 인코딩 하는 법 아래 코드와 주석으로 설명하겠습니다. #파이토치를 불러옵니다 import torch #어떤 데이터의 레이블이 아..
2023. 10. 29.