Skip to content
Permalink
Browse files

adding fit! parameter to MMCTM allowing covariance to be fixed

  • Loading branch information...
funnell committed Nov 6, 2018
1 parent 5e5933e commit e6fc605ecd773f0dfe83a9d2e5fb19c2ea3e1f05
Showing with 6 additions and 2 deletions.
  1. +6 −2 src/MMCTM.jl
@@ -454,7 +454,9 @@ function fitdoc!(model::MMCTM, d::Int)
update_λ!(model, d)
end

function fit!(model::MMCTM; maxiter=100, tol=1e-4, verbose=true, autoα=false)
function fit!(model::MMCTM; maxiter=100, tol=1e-4, verbose=true, autoα=false,
updateΣ=true)

ll = Vector{Float64}[]

for iter in 1:maxiter
@@ -463,7 +465,9 @@ function fit!(model::MMCTM; maxiter=100, tol=1e-4, verbose=true, autoα=false)
end

update_μ!(model)
update_Σ!(model)
if updateΣ
update_Σ!(model)
end
update_γ!(model)
if autoα
update_α!(model)

0 comments on commit e6fc605

Please sign in to comment.
You can’t perform that action at this time.