tmnt 0.7.44b20240124__py3-none-any.whl → 0.7.44b20240125__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. tmnt/modeling.py +3 -1
  2. {tmnt-0.7.44b20240124.dist-info → tmnt-0.7.44b20240125.dist-info}/METADATA +1 -1
  3. {tmnt-0.7.44b20240124.dist-info → tmnt-0.7.44b20240125.dist-info}/RECORD +7 -7
  4. {tmnt-0.7.44b20240124.dist-info → tmnt-0.7.44b20240125.dist-info}/LICENSE +0 -0
  5. {tmnt-0.7.44b20240124.dist-info → tmnt-0.7.44b20240125.dist-info}/NOTICE +0 -0
  6. {tmnt-0.7.44b20240124.dist-info → tmnt-0.7.44b20240125.dist-info}/WHEEL +0 -0
  7. {tmnt-0.7.44b20240124.dist-info → tmnt-0.7.44b20240125.dist-info}/top_level.txt +0 -0
tmnt/modeling.py CHANGED
@@ -493,7 +493,9 @@ class BaseSeqBowVED(BaseVAE):
493
493
  if pre_trained_embedding is not None:
494
494
  self.embedding = nn.Linear(len(pre_trained_embedding.idx_to_vec),
495
495
  pre_trained_embedding.idx_to_vec[0].size, bias=False)
496
- self.apply(self._init_weights)
496
+ #self.apply(self._init_weights)
497
+ self.latent_distribution.apply(self._init_weights)
498
+ self.decoder.apply(self._init_weights)
497
499
 
498
500
  def _init_weights(self, module):
499
501
  if isinstance(module, torch.nn.Linear):
{tmnt-0.7.44b20240124.dist-info → tmnt-0.7.44b20240125.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: tmnt
3
- Version: 0.7.44b20240124
3
+ Version: 0.7.44b20240125
4
4
  Summary: Topic modeling neural toolkit
5
5
  Home-page: https://github.com/mitre/tmnt.git
6
6
  Author: The MITRE Corporation
{tmnt-0.7.44b20240124.dist-info → tmnt-0.7.44b20240125.dist-info}/RECORD RENAMED
@@ -5,7 +5,7 @@ tmnt/distribution.py,sha256=Pmyc5gwDd_-jP7vLVb0vdNQaSSvF1EuiTZEWg3KfmI8,10866
5
5
  tmnt/estimator.py,sha256=IIXjtKB09qUqL_lDiDbhd5IVsW7hLuCHo82fF27xp64,77942
6
6
  tmnt/eval_npmi.py,sha256=ODRDMsBgDM__iCNEX399ck7bAhl7ydvgDqmpfR7Y-q4,5048
7
7
  tmnt/inference.py,sha256=Sw7GO7QiWVEtbPJKBjFB7AiKRmUOZbFZn3tCrsStzWw,17845
8
- tmnt/modeling.py,sha256=_8sQ-OV2QF4tYBs33vdqarf5psbz6BH99L4UxoX9EYw,32840
8
+ tmnt/modeling.py,sha256=Q-CSN0oaftf6RhM3Y3zKk4xw1Wd_WeZmPexZy8nk2Nw,32947
9
9
  tmnt/preprocess/__init__.py,sha256=gwMejkQrnqKS05i0JVsUru2hDUR5jE1hKC10dL934GU,170
10
10
  tmnt/preprocess/tokenizer.py,sha256=-ZgowfbHrM040vbNTktZM_hdl6HDTqxSJ4mDAxq3dUs,14050
11
11
  tmnt/preprocess/vectorizer.py,sha256=RkdivqP76qAJDianV09lONad9NbfBVWLZgIbU_P1-zo,15796
@@ -17,9 +17,9 @@ tmnt/utils/ngram_helpers.py,sha256=VrIzou2oQHCLBLSWODDeikN3PYat1NqqvEeYQj_GhbA,1
17
17
  tmnt/utils/pubmed_utils.py,sha256=3sHwoun7vxb0GV-arhpXLMUbAZne0huAh9xQNy6H40E,1274
18
18
  tmnt/utils/random.py,sha256=qY75WG3peWoMh9pUyCPBEo6q8IvkF6VRjeb5CqJOBF8,327
19
19
  tmnt/utils/recalibrate.py,sha256=TmpB8An8bslICZ13UTJfIvr8VoqiSedtpHxec4n8CHk,1439
20
- tmnt-0.7.44b20240124.dist-info/LICENSE,sha256=qFZJrfJ7Zi4IXDiyiGVrHWic_l1h2tc36tI8Z7rK9bs,11356
21
- tmnt-0.7.44b20240124.dist-info/METADATA,sha256=oqJFAlFCAg-FL_qf9NEo5BJODUGWfDZmXK5z9ghjgp4,1403
22
- tmnt-0.7.44b20240124.dist-info/NOTICE,sha256=p0kYIVAkReTFaGb4C-qPa7h5ztze6hGzOpjCMMbOipU,425
23
- tmnt-0.7.44b20240124.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
24
- tmnt-0.7.44b20240124.dist-info/top_level.txt,sha256=RpYgUl187sXnqmiwKjZZdcDlHz2AALs6bGdUcukyd_E,5
25
- tmnt-0.7.44b20240124.dist-info/RECORD,,
20
+ tmnt-0.7.44b20240125.dist-info/LICENSE,sha256=qFZJrfJ7Zi4IXDiyiGVrHWic_l1h2tc36tI8Z7rK9bs,11356
21
+ tmnt-0.7.44b20240125.dist-info/METADATA,sha256=0duXA_NTiacN4bKgC10fnqMdPeOfVEHqy9EDz7EqquU,1403
22
+ tmnt-0.7.44b20240125.dist-info/NOTICE,sha256=p0kYIVAkReTFaGb4C-qPa7h5ztze6hGzOpjCMMbOipU,425
23
+ tmnt-0.7.44b20240125.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
24
+ tmnt-0.7.44b20240125.dist-info/top_level.txt,sha256=RpYgUl187sXnqmiwKjZZdcDlHz2AALs6bGdUcukyd_E,5
25
+ tmnt-0.7.44b20240125.dist-info/RECORD,,