Merge remote-tracking branch 'ricardoV94/marginalize_hmm' into marginalize_hmm

jessegrabowski · jessegrabowski · commit 8f9ae4058f85 · 2023-11-21T22:05:41.000+01:00
diff --git a/pymc_experimental/marginal_model.py b/pymc_experimental/marginal_model.py
@@ -536,7 +536,7 @@ def eval_logp(x):
         chain_dummy: pt.moveaxis(pt.broadcast_to(domain, (*values[0].shape, domain.size)), -1, 0)
     }
 
-    # This is a (k, T) matrix of logp terms, one for each state - observation pair
+    # This is a (k, T) matrix of logp terms, one for each state - emission pair
     vec_logp_emission = vectorize_graph(tuple(logp_value_dict.values()), sub_dict)
 
     # Step 2: Compute the transition probabilities

Original file line number	Diff line number	Diff line change
`@@ -536,7 +536,7 @@ def eval_logp(x):`
`536`	`536`	`chain_dummy: pt.moveaxis(pt.broadcast_to(domain, (*values[0].shape, domain.size)), -1, 0)`
`537`	`537`	`}`
`538`	`538`
`539`		`- # This is a (k, T) matrix of logp terms, one for each state - observation pair`
	`539`	`+ # This is a (k, T) matrix of logp terms, one for each state - emission pair`
`540`	`540`	`vec_logp_emission = vectorize_graph(tuple(logp_value_dict.values()), sub_dict)`
`541`	`541`
`542`	`542`	`# Step 2: Compute the transition probabilities`