关注
KL散度,提示越界,通过率66.67,找了半天没找出问题在哪里 from math import log
if __name__ == "__main__":
p_data = [int(i) for i in input().split()]
q_data = [int(i) for i in input().split()]
p_total = len(p_data) + 1e-5
q_total = len(q_data) + 1e-5
x_unique = []
for x in p_data + q_data:
if x not in x_unique:
x_unique.append(x)
p_stat = {x: 0 for x in x_unique}
q_stat = {x: 0 for x in x_unique}
for x in p_data:
p_stat[x] += 1
for x in q_data:
q_stat[x] += 1
kl_val = 0
for x in x_unique:
p_proba = p_stat[x] / p_total + 1e-5
q_proba = q_stat[x] / q_total + 1e-5
kl_val += p_proba * log(p_proba / q_proba, 2)
print("%.2f" % kl_val)
查看原帖
点赞 1
相关推荐
投递众安保险等公司9个岗位 >
点赞 评论 收藏
转发
不愿透露姓名的神秘牛友
04-05 09:23
点赞 评论 收藏
转发
牛客热帖
正在热议
# 牛客帮帮团来啦!有问必答 #
710311次浏览 11527人参与
# 许愿池 #
77195次浏览 1542人参与
# 通信硬件人笔面经互助 #
107744次浏览 2178人参与
# 你的秋招进展怎么样了 #
500863次浏览 13424人参与
# 找工作时遇到的神仙HR #
177652次浏览 1744人参与
# 如何写一份好简历 #
259298次浏览 3918人参与
# 铜五铁六真的存在吗? #
27340次浏览 293人参与
# 找工作,你会甘心进小厂还是猛冲大厂 #
35045次浏览 352人参与
# 产品实习,你更倾向大公司or小公司 #
35949次浏览 548人参与
# 非技术岗是怎么找实习的 #
73861次浏览 1385人参与
# 市场营销面经 #
4547次浏览 125人参与
# 互联网公司评价 #
79558次浏览 1087人参与
# 通信硬件薪资爆料 #
196305次浏览 1759人参与
# 你的秋招进行到哪一步了 #
352986次浏览 6269人参与
# 硬件兄弟们 甩出你的华为奖状 #
27511次浏览 180人参与
# 无实习如何秋招上岸 #
224680次浏览 3518人参与
# 投了多少份简历才上岸 #
56661次浏览 947人参与
# 面试中的破防瞬间 #
82565次浏览 1015人参与
# 通信/硬件的薪资开多少,才值得去? #
10740次浏览 140人参与
# 产品人求职现状 #
50589次浏览 747人参与