使用阈值将倾向转换为二进制分类器

时间:2018-11-01 04:36:58

标签: python-3.x numpy scikit-learn

我有一个数组p_array,它存储目标列表为0或1的概率。p_array [0,1]是target [0]为1的概率。我正在尝试找出一种方法来找到p_array [:,1]的阈值,这样我可以准确地预测目标值,例如50%的时间。那就是我想为p_array [:,1]选择一个概率,将上面的所有内容都设为1,将下面的所有内容都设为0,以使其至少在50%的时间内与目标匹配。任何提示,不胜感激。

p_array

array([[0.81982748, 0.18017252],
       [0.80761044, 0.19238956],
       [0.90979368, 0.09020632]])


target

[0,1,0]

0 个答案:

没有答案