cindex的计算加速(numpy)

原公式

def CI(P, Y):
    pair = 0
    summ = 0
    for i in range(1, len(Y)):
        for j in range(0, i):
            if i != j:
                if(Y[i] > Y[j]):
                    pair +=1
                    summ +=  1* (P[i] > P[j]) + 0.5 * (P[i] == P[j])
    return summ/pair

新公式

def CI(P, Y):
    print(P.shape)
    pair = 0
    summ = 0
    temp1 = [(y > Y)  for y in Y]
    temp2 = [(p > P) + (p == P) * 0.5 for p in P]
    summ = 0
    for i, t in enumerate(temp2):
        summ += np.sum(t[:i][temp1[i][:i]])
        pair += np.sum(temp1[i][:i])
 
    return summ/pair

效果:1h -> 0.5min 鸟枪变大炮!!!!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值