tmnt 0.7.44b20240120__py3-none-any.whl → 0.7.44b20240122__py3-none-any.whl

tmnt/distribution.py

@@ -83,7 +83,7 @@ class GaussianDistribution(BaseDistribution):
         z = self.post_sample_dr_o(z)
         return z, KL
     
-    def get_mu_encoding(self, data, include_bn=False):
+    def get_mu_encoding(self, data, include_bn=True, normalize=False):
         """Provide the distribution mean as the natural result of running the full encoder
         
         Parameters:
@@ -94,7 +94,8 @@ class GaussianDistribution(BaseDistribution):
         enc = self.mu_encoder(data)
         if include_bn:
             enc = self.mu_bn(enc)
-        return self.softplus(enc)
+        mu = self.softplus(enc) if normalize else enc
+        return mu
         
 
 
@@ -126,7 +127,7 @@ class GaussianUnitVarDistribution(BaseDistribution):
         KL = self._get_kl_term(mu_bn)
         return self.post_sample_dr_o(z), KL
     
-    def get_mu_encoding(self, data, include_bn=False):
+    def get_mu_encoding(self, data, include_bn=True, normalize=False):
         """Provide the distribution mean as the natural result of running the full encoder
         
         Parameters:
@@ -137,7 +138,8 @@ class GaussianUnitVarDistribution(BaseDistribution):
         enc = self.mu_encoder(data)
         if include_bn:
             enc = self.mu_bn(enc)
-        return self.softplus(enc)
+        mu = self.softplus(enc) if normalize else enc 
+        return mu
         
 
 class LogisticGaussianDistribution(BaseDistribution):
@@ -182,7 +184,7 @@ class LogisticGaussianDistribution(BaseDistribution):
         z = self.post_sample_dr_o(z_p)
         return self.softmax(z), KL
 
-    def get_mu_encoding(self, data, include_bn=False):
+    def get_mu_encoding(self, data, include_bn=True, normalize=False):
         """Provide the distribution mean as the natural result of running the full encoder
         
         Parameters:
@@ -193,7 +195,8 @@ class LogisticGaussianDistribution(BaseDistribution):
         enc = self.mu_encoder(data)
         if include_bn:
             enc = self.mu_bn(enc)
-        return self.softmax(enc)
+        mu = self.softmax(enc) if normalize else enc
+        return mu
         
     
 class VonMisesDistribution(BaseDistribution):
@@ -220,7 +223,7 @@ class VonMisesDistribution(BaseDistribution):
         kld = self.kld_v.expand(batch_size)
         return z_p, kld
 
-    def get_mu_encoding(self, data, include_bn=False):
+    def get_mu_encoding(self, data, include_bn=True, normalize=False):
         """Provide the distribution mean as the natural result of running the full encoder
         
         Parameters:
@@ -231,7 +234,8 @@ class VonMisesDistribution(BaseDistribution):
         enc = self.mu_encoder(data)
         if include_bn:
             enc = self.mu_bn(enc)
-        return self.softplus(enc)
+        mu = self.softplus(enc) if normalize else enc
+        return mu
         
     
 
@@ -247,7 +251,7 @@ class Projection(BaseDistribution):
         kld = torch.zeros(batch_size).to(self.device)
         return mu_bn, kld
 
-    def get_mu_encoding(self, data, include_bn=False):
+    def get_mu_encoding(self, data, include_bn=True, normalize=False):
         """Provide the distribution mean as the natural result of running the full encoder
         
         Parameters:

{tmnt-0.7.44b20240120.dist-info → tmnt-0.7.44b20240122.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tmnt
-Version: 0.7.44b20240120
+Version: 0.7.44b20240122
 Summary: Topic modeling neural toolkit
 Home-page: https://github.com/mitre/tmnt.git
 Author: The MITRE Corporation

{tmnt-0.7.44b20240120.dist-info → tmnt-0.7.44b20240122.dist-info}/RECORD

@@ -1,7 +1,7 @@
 tmnt/__init__.py,sha256=EPNq1H7UMyMewWT_zTGBaC7ZouvCywX_gMX4G1dtmvw,250
 tmnt/configuration.py,sha256=P8PEhzVPKO5xG0FrdTLRQ60OYWigbzPY-OSx_hzQlrY,10054
 tmnt/data_loading.py,sha256=_NpAwmpeFBoQp7xtWOLb6i3WS271JoSJqDx9BMrXtKM,18207
-tmnt/distribution.py,sha256=vNvBq7vLwLOqic-r8knD_pqjnUGKiBmmwbhbwOvl-cE,10633
+tmnt/distribution.py,sha256=Pmyc5gwDd_-jP7vLVb0vdNQaSSvF1EuiTZEWg3KfmI8,10866
 tmnt/estimator.py,sha256=xk4QATqqD8ukxtraOQ6BvSJrdqGTQvX52fNdcgfQ3w8,77801
 tmnt/eval_npmi.py,sha256=ODRDMsBgDM__iCNEX399ck7bAhl7ydvgDqmpfR7Y-q4,5048
 tmnt/inference.py,sha256=Sw7GO7QiWVEtbPJKBjFB7AiKRmUOZbFZn3tCrsStzWw,17845
@@ -17,9 +17,9 @@ tmnt/utils/ngram_helpers.py,sha256=VrIzou2oQHCLBLSWODDeikN3PYat1NqqvEeYQj_GhbA,1
 tmnt/utils/pubmed_utils.py,sha256=3sHwoun7vxb0GV-arhpXLMUbAZne0huAh9xQNy6H40E,1274
 tmnt/utils/random.py,sha256=qY75WG3peWoMh9pUyCPBEo6q8IvkF6VRjeb5CqJOBF8,327
 tmnt/utils/recalibrate.py,sha256=TmpB8An8bslICZ13UTJfIvr8VoqiSedtpHxec4n8CHk,1439
-tmnt-0.7.44b20240120.dist-info/LICENSE,sha256=qFZJrfJ7Zi4IXDiyiGVrHWic_l1h2tc36tI8Z7rK9bs,11356
-tmnt-0.7.44b20240120.dist-info/METADATA,sha256=GZZj0rVgcjfff-EtNd-AUJV-_bN6E3nE57smSxI1sHo,1403
-tmnt-0.7.44b20240120.dist-info/NOTICE,sha256=p0kYIVAkReTFaGb4C-qPa7h5ztze6hGzOpjCMMbOipU,425
-tmnt-0.7.44b20240120.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-tmnt-0.7.44b20240120.dist-info/top_level.txt,sha256=RpYgUl187sXnqmiwKjZZdcDlHz2AALs6bGdUcukyd_E,5
-tmnt-0.7.44b20240120.dist-info/RECORD,,
+tmnt-0.7.44b20240122.dist-info/LICENSE,sha256=qFZJrfJ7Zi4IXDiyiGVrHWic_l1h2tc36tI8Z7rK9bs,11356
+tmnt-0.7.44b20240122.dist-info/METADATA,sha256=ATwYUpYO65LnsfQagNNk0n_tfXsrfq5-Me1Ojg-5jVI,1403
+tmnt-0.7.44b20240122.dist-info/NOTICE,sha256=p0kYIVAkReTFaGb4C-qPa7h5ztze6hGzOpjCMMbOipU,425
+tmnt-0.7.44b20240122.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+tmnt-0.7.44b20240122.dist-info/top_level.txt,sha256=RpYgUl187sXnqmiwKjZZdcDlHz2AALs6bGdUcukyd_E,5
+tmnt-0.7.44b20240122.dist-info/RECORD,,