Label Encoding, One-Hot Encoding / Scikit-learn 레이블 인코딩, 원핫인코딩

Label Encoding

from sklearn.preprocessing import LabelEncoder
le = LabelEncoder()

# 데이터에서 특징 찾기(범주의 수
le.fit(label)

#데이터 변환
label_encoded_label = le.transfrom(label)

 다시 원래대로 바꿀 수 있는 inverse_transform도 있따.

 

One-hot Encoding

from sklearn.preprocessing import OneHotEncoder
ohe = OneHotEncdoer()


# 특징 찾고 데이터 변환
one_fit_transform(label)

# 특징 찾기
ohe.fit(data)

# 데이터 변환
one_hot_encoded = ohe.transform(data)