파이썬에서 다중 레이블 분류를 수행하고 있으며 (sklearn 라이브러리 사용) 다음과 같이 배열 형식의 predict_proba () 함수에서 예측 된 각 레이블에 대해 상위 5 개 클래스를 어떻게받을 수 있는지 궁금합니다.
y_pred=[[1,3,4,5,2],[4,3,2,1,5],[1,2,3,5,4]]
어떻게 할 수 있습니까?
가지고있는 코드를 재현해야 할 수도 있습니다. 다음은 몇 가지 가정을 기반으로 한 솔루션입니다.
당신이 저장할 수 있다면
predict_proba ()
사용하는 시리즈로,
pd.Series (fit.predict_proba (테스트 데이터) [0])
당신은 아마 사용할 수 있습니다
predict_proba (테스트 데이터) .nlargest (n = 5)
자동으로 정렬 된 색인으로 상위 5 개의 확률을 결정합니다.
이 기사는 인터넷에서 수집됩니다. 재 인쇄 할 때 출처를 알려주십시오.
침해가 발생한 경우 연락 주시기 바랍니다[email protected] 삭제
몇 마디 만하겠습니다