[英]Sklearn precision recall curve pos_label for unbalanced dataset which class probability to use
[英]precision recall pos_label python for one-class
目标 :获得一类的 precision
和recall
( y_true
= 1
)
背景 :我检查了http://scikit-learn.org/stable/modules/generated/sklearn.metrics.precision_recall_curve.html#sklearn.metrics.precision_recall_curve ,它指出pos_label
是positive class
的标签,并将其设置为默认为1
。
问题 :
1)如果我只想对positive class
进行precision
和recall
(在这种情况下为y_true
= 1
),我应保持pos_label
= 1
还是应将其更改为pos_label = 0
?
2)还是有更好的方法实现我的目标 ?
下面我在pos_label
= 0
时显示代码
import numpy as np
from sklearn.metrics import precision_recall_fscore_support
y_true = np.array(['0', '1', '1', '0', '1'])
y_pred = np.array(['1', '0', '1', '0', '1'])
out = precision_recall_fscore_support(y_true, y_pred, average='weighted', pos_label = 0)
import numpy as np
from sklearn.metrics import precision_recall_fscore_support
y_true = np.array(['0', '1', '1', '0', '1'])
y_pred = np.array(['1', '0', '1', '0', '1'])
#keep 1's
y_true, y_pred = zip(*[[ytrue[i], ypred[i]] for i in range(len(ytrue)) if ytrue[i]=="1"])
out = precision_recall_fscore_support(y_true, y_pred, average='micro')
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.