다중분류 확률값 제출에 대해서 질문드립니다!
안녕하세요 선생님 계속 질문드려서 죄송합니다. 다름이 아니라 pred_proba를 test 데이터에 적용하고 제출할때 오류가 발생해서 질문드립니다.
y_val_dumm = pd.get_dummies(y_val)
pred_proba = model.predict_proba(X_val)
print(roc_auc_score(y_val_dumm, pred_proba, multi_class = 'ovo'))
####마지막 제출!!
pred_proba = model.predict_proba(test)
pd.DataFrame({'id':test_id,
'Seg':pred_proba})
"Per-column arrays must each be 1-dimensional" 오류가 발생합니다. pred_proba의 컬럼을 지정 안해줘서 그런거 같은데 마지막 제출은 어떻게 해야 하나요??
답변 1
0
좋은 지적해주셨어요! 다중분류 확률은 출제 가능성은 낮지만 나온다면
각 클래스에 대한 확률을 각각의 컬럼으로 만들어야 합니다. 예를 들어, 클래스가 A, B, C 세 가지라면 A에 대한 확률, B에 대한 확률, C에 대한 확률을 각각의 컬럼으로 만들어야 해요
반복문을 사용하는 방법도 있지만 단순하게는 이런식으로 될 수 있겠네요!
각 클래스의 확률을 데이터프레임의 컬럼으로 만들기
pred_proba = model.predict_proba(test)
result_df = pd.DataFrame()
result_df['id'] = test_id #test_id라는 변수에는 test 데이터의 id가 있다는 가정
result_df['prob_class_0'] = pred_proba[:, 0]
result_df['prob_class_1'] = pred_proba[:, 1]
...(컬럼의 수 만큼)0
pred_proba = model.predict_proba(test)
pd.DataFrame({
'id':test_id,
'pred_1':pred_proba[:,0],
'pred_2':pred_proba[:,1],
'pred_3':pred_proba[:,2]
}).to_csv('11111.csv', index = False)이런 식으로 제출하면 되는건가요??
작업형2 카테고리
0
7
1
110강 회귀_8번 문의
0
8
2
XGBRegressor 학습 시 warning 출력
0
13
2
데이터 프레임을 변경해도 되나요?
0
10
2
10회 기출 작업형3 문제 2-1 질문
0
15
2
작업형 2 질문
0
15
2
작업형 2 템플릿 확인 요청
0
13
2
작업형2 인코딩 질문
0
22
2
작업형 3 벼락치기 공부방법 질문
0
30
2
기출 11회 작업형 2_전체 데이터 학습 여부
0
31
2
예측값 결과 소수점 차이
0
28
2
기출 문제와 실전챌린지 연습문제 무엇부터 푸는게 나은가요?
0
35
1
전처리 train() test([ ])
0
23
2
작업형 1 배경지식 질문
0
32
2
옳게 풀은건지 질문드립니다!
0
22
1
roc_auc_score
0
29
2
추가질문 합니다
0
22
2
시험환경 구름
0
20
2
2유형 질문드려요
0
21
2
RandomForest vs lgb
0
28
2
전처리 관련질문
0
29
3
작업형3 기출
0
20
2
유형2에서 데이터분할 생략 가능여부
0
33
2
9회 기출 유형3 질문
0
21
2