tmnt 0.7.0b20231226__py3-none-any.whl → 0.7.4b20231227__py3-none-any.whl

tmnt/estimator.py

@@ -1275,8 +1275,8 @@ class SeqBowEstimator(BaseEstimator):
             vectorizer_file = os.path.join(model_dir, ('vectorizer.pkl' + suffix))
             with open(vectorizer_file, 'wb') as f:
                 pickle.dump(vectorizer, f)
-            with open(conf_file, 'w') as f:
-                f.write(specs)
+        with open(conf_file, 'w') as f:
+            f.write(specs)
         torch.save(self.vocabulary, vocab_file)
 
 

tmnt/inference.py

@@ -65,7 +65,6 @@ class BaseInferencer(object):
     def get_pyldavis_details(self, sp_vec_file_or_X, y=None):
         w_pr, dt_matrix, doc_lengths, term_cnts = self.get_model_details(sp_vec_file_or_X, y=y)
         d1 = w_pr.cpu().detach().numpy().tolist()
-        print("dt_matrix type = {}".format(type(dt_matrix)))
         d2 = list(map(lambda x: x.tolist(), dt_matrix))
         doc_lengths = np.array(doc_lengths)
         d3 = list(doc_lengths.squeeze())

{tmnt-0.7.0b20231226.dist-info → tmnt-0.7.4b20231227.dist-info}/METADATA

@@ -1,16 +1,18 @@
 Metadata-Version: 2.1
 Name: tmnt
-Version: 0.7.0b20231226
+Version: 0.7.4b20231227
 Summary: Topic modeling neural toolkit
 Home-page: https://github.com/mitre/tmnt.git
 Author: The MITRE Corporation
 Author-email: wellner@mitre.org
 License: Apache
-Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
-Requires-Python: >=3.7
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+License-File: NOTICE
 Requires-Dist: optuna
 Requires-Dist: mantichora >=0.9.5
 Requires-Dist: transformers[torch]
@@ -30,6 +32,12 @@ Requires-Dist: tabulate >=0.8.7
 Requires-Dist: torch >=1.13.0
 Requires-Dist: torchtext >=0.13.0
 
-UNKNOWN
+The Topic Modeling Neural Toolkit (TMNT) is a software library that enables training
+topic models as neural network-based variational auto-encoders.
 
+Current stable version is: 0.7.04
+
+Documentation can be found here: https://tmnt.readthedocs.io/en/stable/
+
+The latest development version documentation is here: https://tmnt.readthedocs.io/en/latest
 

{tmnt-0.7.0b20231226.dist-info → tmnt-0.7.4b20231227.dist-info}/RECORD

@@ -4,9 +4,9 @@ tmnt/common_params.py,sha256=uNWs1UuaTx6xlxbS5LailnuXhuIKCg8kCJqxes-kAGY,2547
 tmnt/configuration.py,sha256=P8PEhzVPKO5xG0FrdTLRQ60OYWigbzPY-OSx_hzQlrY,10054
 tmnt/data_loading.py,sha256=Fnn3Pdrw16e6IR_QEPusiUfSCrHlk-3ddKeyzQW_5JE,18569
 tmnt/distribution.py,sha256=JrJe2HaF2uub0S8RxLAjSykg_AF6atvgUhNWbbaxaMo,8382
-tmnt/estimator.py,sha256=lMcuVvPIBiI1ChUpllCFR6ygWnclpcOcYA-ht5r4W3s,78680
+tmnt/estimator.py,sha256=RPJwSYswbhFP0iuINUDp4xNIAB8IuUpXQ1ZRnoq8Xyg,78672
 tmnt/eval_npmi.py,sha256=ODRDMsBgDM__iCNEX399ck7bAhl7ydvgDqmpfR7Y-q4,5048
-tmnt/inference.py,sha256=HLc_lHLoDlTdswvxlQV5iuhbj00dvDlDczEtBftCsJc,18065
+tmnt/inference.py,sha256=k-_EVVFPIGiQl8UQ8bCwcwJVuMRMwuZm7IbmpDl89c8,18004
 tmnt/modeling.py,sha256=Zw6F3PZcaJsUZVuyk-sQIMNWMaqcjQH5cetZTmcqI7g,32940
 tmnt/selector.py,sha256=DWJlbdWKNxJmyLI9IRxCa3FDmaCilxTDzNfIf5mpBqc,9578
 tmnt/trainer.py,sha256=xaJtU_vHAPbos9q86NNdnwz7kpUF5BxGyTWbDG_NBA0,25802
@@ -30,9 +30,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.0b20231226.dist-info/LICENSE,sha256=qFZJrfJ7Zi4IXDiyiGVrHWic_l1h2tc36tI8Z7rK9bs,11356
-tmnt-0.7.0b20231226.dist-info/METADATA,sha256=zgkvGZa_Uhm_zHyRJdmPHyXUIC2mFQ8EpKYyKHqU-dw,997
-tmnt-0.7.0b20231226.dist-info/NOTICE,sha256=p0kYIVAkReTFaGb4C-qPa7h5ztze6hGzOpjCMMbOipU,425
-tmnt-0.7.0b20231226.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-tmnt-0.7.0b20231226.dist-info/top_level.txt,sha256=RpYgUl187sXnqmiwKjZZdcDlHz2AALs6bGdUcukyd_E,5
-tmnt-0.7.0b20231226.dist-info/RECORD,,
+tmnt-0.7.4b20231227.dist-info/LICENSE,sha256=qFZJrfJ7Zi4IXDiyiGVrHWic_l1h2tc36tI8Z7rK9bs,11356
+tmnt-0.7.4b20231227.dist-info/METADATA,sha256=ez5qUHEypKqegmgiPTAN9jP6TdQ1HVgf7xGvBZXpolM,1403
+tmnt-0.7.4b20231227.dist-info/NOTICE,sha256=p0kYIVAkReTFaGb4C-qPa7h5ztze6hGzOpjCMMbOipU,425
+tmnt-0.7.4b20231227.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+tmnt-0.7.4b20231227.dist-info/top_level.txt,sha256=RpYgUl187sXnqmiwKjZZdcDlHz2AALs6bGdUcukyd_E,5
+tmnt-0.7.4b20231227.dist-info/RECORD,,