Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
263 views
in Technique[技术] by (71.8m points)

machine learning - How to prepare confusion matrix from the predicted class probabilities?

There is a Naive Bayesian classifier which is created with a given training data. In the table, the predicted positive class probabilities and the actual class labels are shown. I want to prepare the confusion matrix but I could not find out how to do it with just knowing the probabilities.

ID Actual class label Predicted positive class probability
1 + 0.6
2 + 0.8
3 - 0.2
4 + 0.3
5 - 0.4
question from:https://stackoverflow.com/questions/65927584/how-to-prepare-confusion-matrix-from-the-predicted-class-probabilities

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Reply

0 votes
by (71.8m points)

First, you need to have discrete class labels to compute confusion matrix. Define a threshold on the predicted positive class probability to predict class labels (y_pred). You can then use actual class labels (y_actual) and y_pred to compute the confusion matrix.

from sklearn.metrics import confusion_matrix
confusion_matrix(y_actual, y_pred)

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
OGeek|极客中国-欢迎来到极客的世界,一个免费开放的程序员编程交流平台!开放,进步,分享!让技术改变生活,让极客改变未来! Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...