Update hmm.py

nbhushan · Jul 24, 2023 · c71481c · c71481c
1 parent b351c04
commit c71481c
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/sequence_modelling/hmm.py b/src/sequence_modelling/hmm.py
@@ -69,11 +69,12 @@ def __init__(self, A, O):
         self.logA = np.log(A)
 
     def __repr__(self):
-        n = "" + "nStates: %d\n" % (self.K)
-        a = "" + "A:\n %s" % (str(np.exp(self.logA[:-1])))
-        pi = "" + "\nPi: \n %s" % (str(np.exp(self.logA[-1])))
-        o = "" + "\nObservation distribution: \n %s" % self.O
-        return n + a + pi + o
+        return (
+            f"nStates: {self.K}\n"
+            f"A:\n {self.logA[:-1].round(2)}\n"
+            f"Pi: \n {self.logA[-1].round(2)}\n"
+            f"Observation distribution: \n {self.O}"
+        )
 
     def sample(self, dim=1, N=1000):
         """Generates an observation sequence of length N.