关注
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)
查看原帖
点赞 6
相关推荐
点赞 评论 收藏
分享
10-18 17:42
门头沟学院 产品经理 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 工作中出现了XX情况正常吗 #
2203次浏览 35人参与
# 公司福利里最没用的一项是啥 #
1388次浏览 38人参与
# 回顾今年你干过的最“勇”的一件事 #
1703次浏览 30人参与
# 一人推荐一个值得做的项目 #
1202次浏览 21人参与
# 工作前VS工作后,你的心态变化 #
1406次浏览 17人参与
# 高薪高压 vs 低薪wlb,你怎么选? #
1895次浏览 27人参与
# 毕业季,给职场新人一些建议 #
160762次浏览 2353人参与
# 如何用一句话描述你的职业 #
27506次浏览 176人参与
# 你被哪些公司挂了? #
106063次浏览 652人参与
# 牛友们的论文几号送审 #
61189次浏览 831人参与
# 如何准备秋招 #
67736次浏览 841人参与
# 机械校招之路总结 #
106404次浏览 2035人参与
# 你在职场上见过哪些“水货”同事 #
21461次浏览 150人参与
# 华勤技术工作体验 #
12156次浏览 22人参与
# 你们公司哪个部门最累? #
35357次浏览 246人参与
# 你怎么评价今年的春招? #
146718次浏览 1402人参与
# 哪些公司真双非友好? #
48906次浏览 225人参与
# 职场上哪些事情令人讨厌 #
31308次浏览 131人参与
# 找工作,行业重要还是岗位重要? #
83052次浏览 1665人参与
# 毕业季,你想好怎么跟生活对线了吗? #
259904次浏览 3850人参与
腾讯成长空间 5863人发布