Skip to content

Commit 98c7739

Browse files
Alexey Stukalovalyst
authored andcommitted
EM: move code refs to docstring
1 parent efa9571 commit 98c7739

1 file changed

Lines changed: 6 additions & 7 deletions

File tree

src/observed/EM.jl

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,8 @@
22
### Expectation Maximization Algorithm
33
############################################################################################
44

5-
# An EM Algorithm for MVN-distributed Data with missing values
6-
# Adapted from supplementary Material to the book Machine Learning: A Probabilistic Perspective
7-
# Copyright (2010) Kevin Murphy and Matt Dunham
8-
# found at https://github.com/probml/pmtk3/blob/master/toolbox/BasicModels/gauss/sub/gaussMissingFitEm.m
9-
# and at https://github.com/probml/pmtk3/blob/master/toolbox/Algorithms/optimization/emAlgo.m
10-
115
# what about random restarts?
126

13-
# outer function ---------------------------------------------------------------------------
147
"""
158
em_mvn(;
169
observed::SemObservedMissing,
@@ -21,6 +14,12 @@
2114
2215
Estimates the covariance matrix and mean vector of the normal distribution via expectation maximization for `observed`.
2316
Overwrites the statistics stored in `observed`.
17+
18+
Uses the EM algorithm for MVN-distributed data with missing values
19+
adapted from the supplementary material to the book *Machine Learning: A Probabilistic Perspective*,
20+
copyright (2010) Kevin Murphy and Matt Dunham: see
21+
[*gaussMissingFitEm.m*](https://github.com/probml/pmtk3/blob/master/toolbox/BasicModels/gauss/sub/gaussMissingFitEm.m) and
22+
[*emAlgo.m*](https://github.com/probml/pmtk3/blob/master/toolbox/Algorithms/optimization/emAlgo.m) scripts.
2423
"""
2524
function em_mvn(
2625
observed::SemObservedMissing;

0 commit comments

Comments
 (0)