In this homework, you will implement an expectation-maximization (EM) clustering algorithm in Python. Here are the steps you need to follow:
1. You are given a two-dimensional data set in the file named hw07_data_set.csv, which contains 300 data points generated randomly from five bivariate Gaussian densities with the following parameters.
!
+2.5
!
+0.8
−0.6
!
= 50
"
= #+2.5( ,
Σ
"
= #−0.6
+0.8( ,
−2.5
Σ
+0.8
+0.6
"
= 50
#
= #+2.5( ,
#
= #+0.6
+0.8( ,
−2.5
Σ
+0.8
−0.6
#
= 50
$
= #−2.5( ,
$
= #−0.6
+0.8( ,
+2.5
Σ
+0.8
+0.6
$
= 50
%
= #−2.5( ,
%
= #+0.6
+0.8( ,
+0.0
Σ
+1.6
+0.0
%
= 100
= #+0.0( ,
= #+0.0
+1.6( ,
The given data points are shown in the following figure.
2. To initialize your EM algorithm, you should take the centroids given in the file named hw07_initial_centroids.csv as the initial values for the mean vectors. By assigning the data points to the nearest center, estimate the initial covariance matrices and prior probabilities in your EM algorithm.
3. After the initialization step, run your EM algorithm for 100 iterations. Report the mean vectors your EM algorithm finds. Your results should be similar to the following matrix.
[[-2.44390052 -2.5453942 ]
[ 2.50354332 2.51134859]
[ 2.56726404 -2.55477253]
[ 0.12794677 0.15595776]
[-2.41465305 2.4855615 ]]
4. Draw the clustering result obtained by your EM algorithm by coloring each cluster with a different color. You should also draw the original Gaussian densities you use to generate data points and the Gaussian densities your EM algorithm finds with dashed and solid lines, respectively. Draw these Gaussian densities where their values are equal to 0.05. Your figure should be similar to the following figure.
What to submit: You need to submit your source code in a single file (.py file) and a short report explaining your approach (.doc, .docx, or .pdf file). You will put these two files in a single zip file named as STUDENTID.zip, where STUDENTID should be replaced with your 7-digit student number.
How to submit: Submit the zip file you created to Blackboard. Please follow the exact style mentioned and do not send a zip file named as STUDENTID.zip. Submissions that do not follow these guidelines will not be graded.
Late submission policy: Late submissions will not be graded.
Cheating policy: Very similar submissions will not be graded.
