tmnt 0.7.0b20231226__py3-none-any.whl → 0.7.4b20231227__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.
- tmnt/estimator.py +2 -2
- tmnt/inference.py +0 -1
- {tmnt-0.7.0b20231226.dist-info → tmnt-0.7.4b20231227.dist-info}/METADATA +12 -4
- {tmnt-0.7.0b20231226.dist-info → tmnt-0.7.4b20231227.dist-info}/RECORD +8 -8
- {tmnt-0.7.0b20231226.dist-info → tmnt-0.7.4b20231227.dist-info}/LICENSE +0 -0
- {tmnt-0.7.0b20231226.dist-info → tmnt-0.7.4b20231227.dist-info}/NOTICE +0 -0
- {tmnt-0.7.0b20231226.dist-info → tmnt-0.7.4b20231227.dist-info}/WHEEL +0 -0
- {tmnt-0.7.0b20231226.dist-info → tmnt-0.7.4b20231227.dist-info}/top_level.txt +0 -0
tmnt/estimator.py
CHANGED
@@ -1275,8 +1275,8 @@ class SeqBowEstimator(BaseEstimator):
|
|
1275
1275
|
vectorizer_file = os.path.join(model_dir, ('vectorizer.pkl' + suffix))
|
1276
1276
|
with open(vectorizer_file, 'wb') as f:
|
1277
1277
|
pickle.dump(vectorizer, f)
|
1278
|
-
|
1279
|
-
|
1278
|
+
with open(conf_file, 'w') as f:
|
1279
|
+
f.write(specs)
|
1280
1280
|
torch.save(self.vocabulary, vocab_file)
|
1281
1281
|
|
1282
1282
|
|
tmnt/inference.py
CHANGED
@@ -65,7 +65,6 @@ class BaseInferencer(object):
|
|
65
65
|
def get_pyldavis_details(self, sp_vec_file_or_X, y=None):
|
66
66
|
w_pr, dt_matrix, doc_lengths, term_cnts = self.get_model_details(sp_vec_file_or_X, y=y)
|
67
67
|
d1 = w_pr.cpu().detach().numpy().tolist()
|
68
|
-
print("dt_matrix type = {}".format(type(dt_matrix)))
|
69
68
|
d2 = list(map(lambda x: x.tolist(), dt_matrix))
|
70
69
|
doc_lengths = np.array(doc_lengths)
|
71
70
|
d3 = list(doc_lengths.squeeze())
|
@@ -1,16 +1,18 @@
|
|
1
1
|
Metadata-Version: 2.1
|
2
2
|
Name: tmnt
|
3
|
-
Version: 0.7.
|
3
|
+
Version: 0.7.4b20231227
|
4
4
|
Summary: Topic modeling neural toolkit
|
5
5
|
Home-page: https://github.com/mitre/tmnt.git
|
6
6
|
Author: The MITRE Corporation
|
7
7
|
Author-email: wellner@mitre.org
|
8
8
|
License: Apache
|
9
|
-
Platform: UNKNOWN
|
10
9
|
Classifier: Programming Language :: Python :: 3
|
11
10
|
Classifier: License :: OSI Approved :: Apache Software License
|
12
11
|
Classifier: Operating System :: OS Independent
|
13
|
-
Requires-Python: >=3.
|
12
|
+
Requires-Python: >=3.8
|
13
|
+
Description-Content-Type: text/markdown
|
14
|
+
License-File: LICENSE
|
15
|
+
License-File: NOTICE
|
14
16
|
Requires-Dist: optuna
|
15
17
|
Requires-Dist: mantichora >=0.9.5
|
16
18
|
Requires-Dist: transformers[torch]
|
@@ -30,6 +32,12 @@ Requires-Dist: tabulate >=0.8.7
|
|
30
32
|
Requires-Dist: torch >=1.13.0
|
31
33
|
Requires-Dist: torchtext >=0.13.0
|
32
34
|
|
33
|
-
|
35
|
+
The Topic Modeling Neural Toolkit (TMNT) is a software library that enables training
|
36
|
+
topic models as neural network-based variational auto-encoders.
|
34
37
|
|
38
|
+
Current stable version is: 0.7.04
|
39
|
+
|
40
|
+
Documentation can be found here: https://tmnt.readthedocs.io/en/stable/
|
41
|
+
|
42
|
+
The latest development version documentation is here: https://tmnt.readthedocs.io/en/latest
|
35
43
|
|
@@ -4,9 +4,9 @@ tmnt/common_params.py,sha256=uNWs1UuaTx6xlxbS5LailnuXhuIKCg8kCJqxes-kAGY,2547
|
|
4
4
|
tmnt/configuration.py,sha256=P8PEhzVPKO5xG0FrdTLRQ60OYWigbzPY-OSx_hzQlrY,10054
|
5
5
|
tmnt/data_loading.py,sha256=Fnn3Pdrw16e6IR_QEPusiUfSCrHlk-3ddKeyzQW_5JE,18569
|
6
6
|
tmnt/distribution.py,sha256=JrJe2HaF2uub0S8RxLAjSykg_AF6atvgUhNWbbaxaMo,8382
|
7
|
-
tmnt/estimator.py,sha256=
|
7
|
+
tmnt/estimator.py,sha256=RPJwSYswbhFP0iuINUDp4xNIAB8IuUpXQ1ZRnoq8Xyg,78672
|
8
8
|
tmnt/eval_npmi.py,sha256=ODRDMsBgDM__iCNEX399ck7bAhl7ydvgDqmpfR7Y-q4,5048
|
9
|
-
tmnt/inference.py,sha256=
|
9
|
+
tmnt/inference.py,sha256=k-_EVVFPIGiQl8UQ8bCwcwJVuMRMwuZm7IbmpDl89c8,18004
|
10
10
|
tmnt/modeling.py,sha256=Zw6F3PZcaJsUZVuyk-sQIMNWMaqcjQH5cetZTmcqI7g,32940
|
11
11
|
tmnt/selector.py,sha256=DWJlbdWKNxJmyLI9IRxCa3FDmaCilxTDzNfIf5mpBqc,9578
|
12
12
|
tmnt/trainer.py,sha256=xaJtU_vHAPbos9q86NNdnwz7kpUF5BxGyTWbDG_NBA0,25802
|
@@ -30,9 +30,9 @@ tmnt/utils/ngram_helpers.py,sha256=VrIzou2oQHCLBLSWODDeikN3PYat1NqqvEeYQj_GhbA,1
|
|
30
30
|
tmnt/utils/pubmed_utils.py,sha256=3sHwoun7vxb0GV-arhpXLMUbAZne0huAh9xQNy6H40E,1274
|
31
31
|
tmnt/utils/random.py,sha256=qY75WG3peWoMh9pUyCPBEo6q8IvkF6VRjeb5CqJOBF8,327
|
32
32
|
tmnt/utils/recalibrate.py,sha256=TmpB8An8bslICZ13UTJfIvr8VoqiSedtpHxec4n8CHk,1439
|
33
|
-
tmnt-0.7.
|
34
|
-
tmnt-0.7.
|
35
|
-
tmnt-0.7.
|
36
|
-
tmnt-0.7.
|
37
|
-
tmnt-0.7.
|
38
|
-
tmnt-0.7.
|
33
|
+
tmnt-0.7.4b20231227.dist-info/LICENSE,sha256=qFZJrfJ7Zi4IXDiyiGVrHWic_l1h2tc36tI8Z7rK9bs,11356
|
34
|
+
tmnt-0.7.4b20231227.dist-info/METADATA,sha256=ez5qUHEypKqegmgiPTAN9jP6TdQ1HVgf7xGvBZXpolM,1403
|
35
|
+
tmnt-0.7.4b20231227.dist-info/NOTICE,sha256=p0kYIVAkReTFaGb4C-qPa7h5ztze6hGzOpjCMMbOipU,425
|
36
|
+
tmnt-0.7.4b20231227.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
|
37
|
+
tmnt-0.7.4b20231227.dist-info/top_level.txt,sha256=RpYgUl187sXnqmiwKjZZdcDlHz2AALs6bGdUcukyd_E,5
|
38
|
+
tmnt-0.7.4b20231227.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|