我有预测概率的向量。我想将每个概率与几个阈值进行比较,最后得到矩阵结果[i,j],如果概率[i]大于阈值[j],则保存布尔值。
这就是我到目前为止......
y = np.random.randint(2, size=6)
pred = np.array([0.21,0.37,0.18,0.88,0.54,0.23])
thresh = np.array([0.25, 0.5, 0.75])
pred_true = []
for i in range(thresh.shape[0]): #for each threshold
pred_true.append(pred > thresh[i])
v_pred_true = np.vstack(pred_true).T
v_pred_true
给出:
Out[40]:
array([[False, False, False],
[ True, False, False],
[False, False, False],
[ True, True, True],
[ True, True, False],
[False, False, False]], dtype=bool)
但是我想找到一种比较简单的方法来做到这一点。