Information Gain 计算 python实现

关注

原创最新推荐文章于 2026-08-28 01:39:41 发布 5.4k 阅读 AI 写同款· GEO 优化 ›

分享关于视觉显著性评估指标的一些代码分享关于视觉显著性评估指标的一些代码博客小白兼今年9月准备研一的人工智能小白，第一次发博客，想分享一些之前毕设里用到的视觉显著性评估指标的python代码，包括CC,SIM,AUC,NSS等，要是对大家有一些帮助就最好了。 By：精灵之羽(1032332623)欢迎大神加好友能带飞我呀代码函数主要有： - 线性相关性（英文全称：Linear Correlation Coeffcent，... 阅读详情

from math import log

def I(*args):
    total = sum(args) + 0.0
    result = 0.0
    for i in args:
        if i == 0:
            result += 0
        else:
            result += i / total * log( i / total, 2)
    return -result

#num表示分类的个数
def E(num, *args):
    if len(args) % num != 0:
        print "Error len(args)"
    result = 0.0
    total = sum(args)
    for x in xrange(len(args) / num):
        k = x * num
        total_up = 0.0 + sum(args[k:k + num])
        result += total_up / total * I(*args[k:k + num])
    return result

def Gain(i, e):
    return i - e

#取个别名
i = I
e = E
g = Gain

if __name__ == "__main__":
    #for example
    print i(9, 5)
    print e(2, 2, 3, 4, 0, 3, 2)

收录于