to category

Han Hanju·2021년 10월 15일
0
post-thumbnail
  • label을 숫자로 맵핑시켜줘야하는 경우가 있다. 그럴때 사용!

1. label을 숫자로 변환

df['market_category_name'].astype('category').cat.dode

2. 숫자를 다시 label원본으로 변환

df['market_category_name'].astype('category').cat.categories[arr_pre][0]
  • softmax의 경우 아래와같이 가장 큰 값을 선택 후 해당 숫자로 label원본 변환
def arg_max(data):
    arr_pre = []
    for d in data:
        output = d
        f = lambda i: output[i]
        arr_pre.append(max(range(len(output)), key=f))
    return df['market_category_name'].astype('category').cat.categories[arr_pre][0]
profile
Data Analytics Engineer

0개의 댓글