fixed errors in saliency calculation method

maximtrp · maximtrp · commit 63a21291dfcc · 2025-03-16T00:01:28.000+01:00
diff --git a/src/tmplot/__init__.py b/src/tmplot/__init__.py
@@ -5,4 +5,4 @@
 from ._vis import *  # noqa: F401, F403
 from ._metrics import *  # noqa: F401, F403
 
-__version__ = '0.1.3'
+__version__ = '0.2.0'
diff --git a/src/tmplot/_helpers.py b/src/tmplot/_helpers.py
@@ -319,10 +319,10 @@ def get_salient_terms(phi: ndarray, theta: ndarray) -> ndarray:
     p_w = calc_terms_marg_probs(phi, p_t)
 
     def _p_tw(phi, w, t):
-        return phi[w, t] * p_t[t] / p_w[w]
+        return array(phi)[w, t] * p_t[t] / p_w[w]
 
     saliency = array(
-        (
+        [
             p_w[w]
             * sum(
                 (
@@ -331,7 +331,7 @@ def _p_tw(phi, w, t):
                 )
             )
             for w in range(phi.shape[0])
-        )
+        ]
     )
     # saliency(term w) = frequency(w)
     # * [sum_t p(t | w) * log(p(t | w)/p(t))] for topics t

Original file line number	Diff line number	Diff line change
`@@ -319,10 +319,10 @@ def get_salient_terms(phi: ndarray, theta: ndarray) -> ndarray:`
`319`	`319`	`p_w = calc_terms_marg_probs(phi, p_t)`
`320`	`320`
`321`	`321`	`def _p_tw(phi, w, t):`
`322`		`- return phi[w, t] * p_t[t] / p_w[w]`
	`322`	`+ return array(phi)[w, t] * p_t[t] / p_w[w]`
`323`	`323`
`324`	`324`	`saliency = array(`
`325`		`- (`
	`325`	`+ [`
`326`	`326`	`p_w[w]`
`327`	`327`	`* sum(`
`328`	`328`	`(`
`@@ -331,7 +331,7 @@ def _p_tw(phi, w, t):`
`331`	`331`	`)`
`332`	`332`	`)`
`333`	`333`	`for w in range(phi.shape[0])`
`334`		`- )`
	`334`	`+ ]`
`335`	`335`	`)`
`336`	`336`	`# saliency(term w) = frequency(w)`
`337`	`337`	`# * [sum_t p(t \| w) * log(p(t \| w)/p(t))] for topics t`