实验8: EM算法

实验8: EM算法

一、实验目的

了解EM算法的构建过程和代码实现，应用EM算法解决简单的实际问题。

二、实验准备

安装python和pycharm，了解python基础编程和pycharm使用。

三、实验内容

基于程序中生成的数据，补充完整下面EM算法相关程序，粘贴运行成功的结果截图，认真体会EM算法的实现过程。

'''

数据集：构造数据集（两个高斯分布混合）

数据集长度：1000

----------------------------

'''



import numpy as np

import random

import math



def loadData(mu0, sigma0, mu1, sigma1, alpha0, alpha1):

    '''

    初始化数据集

    这里通过服从高斯分布的随机函数来伪造数据集

    :param mu0: 高斯0的均值

    :param sigma0: 高斯0的方差

    :param mu1: 高斯1的均值

    :param sigma1: 高斯1的方差

    :param alpha0: 高斯0的系数