원핫인코더

juyeon·2022년 10월 24일
0

데이터 다루기

목록 보기
29/35
post-custom-banner

encoder = OneHotEncoder()
encoder.fit(train[['type']])
onehot = encoder.transform(train[['type']])
onehot = onehot.toarray()
onehot = pd.DataFrame(onehot)
onehot.columns = encoder.get_feature_names()
onehot = pd.concat([train, onehot], axis = 1)
train = train.drop(columns = ['type'])

profile
내 인생의 주연

0개의 댓글