fit!

gdalle · Dec 7, 2023 · 547ac29 · 547ac29
1 parent 3a20350
commit 547ac29
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/types/abstract_hmm.jl b/src/types/abstract_hmm.jl
@@ -86,7 +86,13 @@ function obs_logdensities!(logb::AbstractVector, hmm::AbstractHMM, obs, control)
 end
 
 """
-    fit!(hmm, ...)
+    fit!(
+        hmm::AbstractHMM,
+        obs_seq::AbstractVector;
+        control_seq::AbstractVector,
+        seq_ends::AbstractVector{Int},
+        fb_storage::ForwardBackwardStorage,
+    )
 
 Update `hmm` in-place based on information generated during forward-backward.
 """