Fit PSD model with EM algorithm, and use the loss function as a stopping criterion.
Arguments
- G
The I x J matrix of counts; all entries of G should be taken from {0,1,2}.
- K
An integer 2 or greater giving the matrix rank.
- epsilon
Convergence criterion.
- maxiter
The maximum number of iterations.
Value
A List with the following parameters:
P
The population scale matrix of the individuals.
F
The gene scale matrix of the populations.
Loss
A vector represents the value of the loss function which records once for 10 iterations.
Iterations
An integer represents the number of iterations.
Examples
G <- matrix(c(0,0,1, 0,2,1, 1,0,1, 0,1,0, 1,0,0), 3, 5)
psd_fit_em(G, 2, 1e-5, 10)
#> $P
#> [,1] [,2]
#> [1,] 0.98716323 0.01283677
#> [2,] 0.02402906 0.97597094
#> [3,] 0.70213453 0.29786547
#>
#> $F
#> [,1] [,2] [,3] [,4] [,5]
#> [1,] 0.26133691 0.02278575 0.5717675706 1.424245e-11 3.069712e-01
#> [2,] 0.02562684 0.99999886 0.0007086483 4.232162e-01 4.607535e-10
#>
#> $Loss
#> [1] -0.7195729
#>
#> $Iterations
#> [1] 10
#>