From 564e2b8b950b327f256a707dceb84884096566b1 Mon Sep 17 00:00:00 2001
From: Matthew Johnson <mattjj@csail.mit.edu>
Date: Mon, 1 Feb 2016 09:34:31 -0500
Subject: [PATCH] add check for last iteration of forward message passing,
needed for inhomog
---
pyhsmm/internals/hmm_messages.h | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/pyhsmm/internals/hmm_messages.h b/pyhsmm/internals/hmm_messages.h
index f10e9b6..a1f7bbc 100644
--- a/pyhsmm/internals/hmm_messages.h
+++ b/pyhsmm/internals/hmm_messages.h
@@ -154,7 +154,9 @@ namespace hmm
return -numeric_limits<Type>::infinity();
}
#endif
- ein_potential = ealphan.row(t) * eAs.block(t*M*hetero,0,M,M);
+ if (likely(t != T-1)) {
+ ein_potential = ealphan.row(t) * eAs.block(t*M*hetero,0,M,M);
+ }
}
return logtot;
}
--
GitLab