简单EM算法python实现

最新推荐文章于 2021-01-29 19:57:19 发布

ShawDa

最新推荐文章于 2021-01-29 19:57:19 发布

阅读量1.6k

点赞数

本文链接：https://blog.csdn.net/sinat_36811967/article/details/80364721

版权

本文介绍了EM算法的Python实现，并通过实验观察到：当初始概率相等时，最终结果为0.66；当初始概率不等时，会收敛到0.8和0.52的不同比例。

摘要由CSDN通过智能技术生成

reference:https://blog.csdn.net/sinat_36811967/article/details/79801095

# -*- coding:utf-8 -*-
# https://blog.csdn.net/sinat_36811967/article/details/79801095

from __future__ import print_function

# 知道投掷的是a或b,求Coin a和b投掷结果是head的最大似然估计,用来评估EM算法
MLE_a = 24.0 / (24 + 6)
MLE_b = 9.0 / (9 + 11)
print("real:", MLE_a, MLE_b)

# 不知道投掷的是a还是b,用EM算法估计
# init
init_a = 0.6
init_b = 0.5
h1, t1 = 5, 5
h2, t2 =