최근 GPT가 train과 test의 특정 컬럼 유니크 수가 다를 때 사용하는 매우 간단한 방법으로 align을 알려주고 있어 제게 사용해도 괜찮은지 질문이 많이 들어오고 있어요:)
우선 사용해도 됩니다.🤩 역시 GPT
단, 단점이 없는 것은 아니니 장단점을 알고 사용하면 더 좋을 것 같아요 💪💪💪💪
train = pd.get_dummies(train)
test = pd.get_dummies(test)
train, test = train.align(test, join='left', axis=1, fill_value=0)
✅train.align(test, join='left', axis=1, fill_value=0)
train의 컬럼 순서와 구조에 test를 맞추는 코드
💚장점
매우 간단하다
한 줄로train기준에 맞춰test의 컬럼 순서 및 구조를 통일할 수 있다모델 학습 시 오류 방지
train/test 컬럼이 맞지 않아 생기는 오류를 사전에 방지할 수 있다fill_value로 누락된 컬럼 채우기 가능
test에 없는 컬럼을 0으로 자동 채워주는 등 유연한 처리 가능
❤단점
test에만 있는 컬럼은 삭제됨
join='left'이므로 train에 없는 컬럼은 test에서 강제로 제거됨