[에러해결] 'predict_proba is not available when probability=False'

'predict_proba is not available when probability=False'

 

from sklearn.svm import SVC
svm = SVC(kernel='linear', C=1.0, random_state=0)
svm.fit(x_train, y_train)
svm_probs = svm.predict_proba(x_test)

Python에서 sklearn 라이브러리를 이용해 SVM 모델을 만들고 위 코드를 실행했을 때 마지막 줄에서 에러가 발생했다. 

 

해결 방법은 SVC 객체 생성 시 파라미터에 probability=True 를 붙여준다.

svm = SVC(kernel='linear', C=1.0, random_state=0, probability=True)

기본적으로 probability 파라미터는 False로 설정되기 때문에 svm.predict_proba() 함수를 실행하면 에러가 난다. 따라서 True로 명시해준다.

 

 

 

참고 : www.discoverbits.in/371/sklearn-attributeerror-predict_proba-available-probability