[sklearn] One Hot Encoding 되돌리기

Python/Scikit-Learn 2021. 1. 24. 21:56

scikit-learn에서 OneHotEncoder를 사용하고 되돌리는 방법에 대해 알아보자. 머신러닝을 할 때, sklearn.preprocessing.OneHotEncoder 또는 sklearn.preprocessing.LabelEncoder를 사용하여 학습을 하게된다면, 최종 예측 결과도 one-hot encoded 된 결과로 나오게 되는데, 간단하게 inverse_transform을 통해 원래 형식으로 바꿀 수 있다. 아래처럼 샘플 시리즈 ["a", "b", "c"]가 있다고 할 때, OneHotEncoder 또는 LabelEncoder를 사용하게 된다면: import pandas as pd from sklearn.preprocessing import LabelEncoder, OneHotEncod..

Article Thumbnail