tmnt 0.7.60__py3-none-any.whl → 0.7.61__py3-none-any.whl

tmnt/__init__.py

@@ -1,8 +1,9 @@
 # coding: utf-8
 
-from .distribution import *
+
 from .preprocess import *
 from .sparse import *
 from .utils import *
+from .distribution import *
 
-__all__ = distribution.__all__ + preprocess.__all__ + utils.__all__ + sparse.__all__
+__all__ = distribution.__all__ + preprocess.__all__ + utils.__all__ # + sparse.__all__

tmnt/estimator.py

@@ -15,7 +15,7 @@ import numpy as np
 import scipy.sparse as sp
 import json
 
-from sklearn.metrics import average_precision_score, top_k_accuracy_score, roc_auc_score, ndcg_score, precision_recall_fscore_support
+from sklearn.metrics import average_precision_score, top_k_accuracy_score, roc_auc_score, ndcg_score
 from tmnt.data_loading import PairedDataLoader, SingletonWrapperLoader, SparseDataLoader, get_llm_model
 from tmnt.modeling import BowVAEModel, SeqBowVED, BaseVAE
 from tmnt.modeling import CrossBatchCosineSimilarityLoss, GeneralizedSDMLLoss, MultiNegativeCrossEntropyLoss, MetricSeqBowVED, MetricBowVAEModel
@@ -29,18 +29,15 @@ from torcheval.metrics import MultilabelAUPRC, MulticlassAUPRC
 ## huggingface specifics
 from transformers.trainer_pt_utils import get_parameter_names
 from transformers.pytorch_utils import ALL_LAYERNORM_LAYERS
-from transformers.optimization import AdamW, get_scheduler
+from transformers.optimization import get_scheduler
 
 ## model selection
 import optuna
 
-from itertools import cycle
 import pickle
 from typing import List, Tuple, Dict, Optional, Union, NoReturn
 
 import torch
-from torch.utils.data import Dataset, DataLoader
-from tqdm import tqdm
 
 MAX_DESIGN_MATRIX = 250000000
 

tmnt/sparse/config.py

@@ -1,5 +1,7 @@
 import torch 
 
+__all__ = ['get_default_cfg']
+
 def get_default_cfg():
     default_cfg = {
         "seed": 49,

tmnt/sparse/estimator.py

@@ -8,6 +8,8 @@ import io, json
 from tmnt.sparse.modeling import BaseAutoencoder
 from typing import List
 
+__all__ = ['ActivationsStore', 'build_activation_store', 'build_activation_store_batching', 'train_sparse_encoder_decoder']
+
 class ActivationsStore:
     def __init__(
         self,

tmnt/sparse/inference.py

@@ -11,6 +11,7 @@ from datasets.arrow_writer import ArrowWriter
 from tmnt.inference import SeqVEDInferencer
 import io, json
 
+__all__ = ['batch_process_to_arrow']
 
 def csr_to_indices_data(csr_mat):
     return [ (csr_mat.getrow(ri).indices, csr_mat.getrow(ri).data) for ri in range(csr_mat.shape[0]) ]

tmnt/sparse/modeling.py

@@ -3,6 +3,9 @@ import torch.nn as nn
 import torch.nn.functional as F
 import torch.autograd as autograd
 
+
+__all__ = ['BatchTopKEncoder', 'BatchTopKSAE', 'TopKEncoder', 'TopKSAE', 'VanillaEncoder', 'VanillaSAE', 'JumpReLUEncoder', 'JumpReLUSAE']
+
 class BaseEncoder(nn.Module):
     
     def __init__(self, cfg):

{tmnt-0.7.60.dist-info → tmnt-0.7.61.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tmnt
-Version: 0.7.60
+Version: 0.7.61
 Summary: Topic modeling neural toolkit
 Home-page: https://github.com/mitre/tmnt.git
 Author: The MITRE Corporation
@@ -48,7 +48,7 @@ Dynamic: summary
 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.60
+Current stable version is: 0.7.61
 
 Documentation can be found here: https://tmnt.readthedocs.io/en/stable/
 

{tmnt-0.7.60.dist-info → tmnt-0.7.61.dist-info}/RECORD

@@ -1,8 +1,8 @@
-tmnt/__init__.py,sha256=s7YqLj32HKhIYO1QbD0zms8rDlTrleJM8LRjKt8bDPk,200
+tmnt/__init__.py,sha256=3W6sRKacvIMQuxGZpscjU52B4sDcasEA1RCXJVHaK-Q,203
 tmnt/configuration.py,sha256=P8PEhzVPKO5xG0FrdTLRQ60OYWigbzPY-OSx_hzQlrY,10054
 tmnt/data_loading.py,sha256=LcVcXX00UsuAillRPILcvmqj3AsCIgzB6V_S6lfsbIY,19335
 tmnt/distribution.py,sha256=4gn1wnszVAErzICCvZXSYki0G78WC3_jyBr27N-Aj3E,15108
-tmnt/estimator.py,sha256=KnnvSNXm6cRL0GwDrGdgqqPX5ZubpCQ0WqcSXJDkUU4,68072
+tmnt/estimator.py,sha256=SJW9koDHb_lTwmqXm4Ilgga7MMMmRGLd3ylEFYu3bZg,67933
 tmnt/eval_npmi.py,sha256=8S-IE-bEhtQofF6oKeXs7oaUeu-7yDlaEqjMj52gmNQ,6549
 tmnt/inference.py,sha256=Iwc2_w7QrS1epiVEm_Ewx5sYFNNMDfvhMJETOgJqm0E,15783
 tmnt/modeling.py,sha256=rGHQsW7ldycFUd1f9NzcnNuSRElr600vLwmYPl6YY0M,30215
@@ -10,10 +10,10 @@ tmnt/preprocess/__init__.py,sha256=gwMejkQrnqKS05i0JVsUru2hDUR5jE1hKC10dL934GU,1
 tmnt/preprocess/tokenizer.py,sha256=-ZgowfbHrM040vbNTktZM_hdl6HDTqxSJ4mDAxq3dUs,14050
 tmnt/preprocess/vectorizer.py,sha256=RaianZ_DG3Nc-RI96FtmI4PCZPi5Nipx9a5xndLZ52M,20689
 tmnt/sparse/__init__.py,sha256=BEhOm_o0UrVUKTG3rSiBJzE7qQQL9HRSZ1MHCA2GJu8,249
-tmnt/sparse/config.py,sha256=gfJ1BAP3zzMKKUJExrs8D0hHB7XOVvqXPpmx0ECAPIE,796
-tmnt/sparse/estimator.py,sha256=SOEPkQo7T2RuBJLYRAk65MhoYjgYtNCAzlvhqq10c5E,4596
-tmnt/sparse/inference.py,sha256=etOuXTxh8bKc7EoohZLYYufFAD51aEpq8mGj1vjULbg,2813
-tmnt/sparse/modeling.py,sha256=IejHbRXyj5WsEthkVbp3vYF1wTAAsihbneaPuCgAfGA,13612
+tmnt/sparse/config.py,sha256=DzqzmclxbUmmBZHS7poMicsvgOZCMXTXWEY5bTpPrRI,827
+tmnt/sparse/estimator.py,sha256=PSqIopCZG7WO5MXYcN-CAs7R7M36jZXjQeUp0I5kG9Q,4721
+tmnt/sparse/inference.py,sha256=KiSUjo_lO14qkUzzo9HuOWZ5S5oJRzoSsSa2dOg54UY,2850
+tmnt/sparse/modeling.py,sha256=hy2nkI2tq3Bpmj7OfXr_zwHRajWBwsqz7T3wHTuEbyE,13753
 tmnt/utils/__init__.py,sha256=1PZsxRPsHI_DnOpxD0iAhLxhxHnx6Svzg3W-79YfWWs,237
 tmnt/utils/csv2json.py,sha256=A1TXy-uxA4dc9tw0tjiHzL7fv4C6b0Uc_bwI1keTmKU,795
 tmnt/utils/log_utils.py,sha256=ZtR4nF_Iee23ev935YQcTtXv-cCC7lgXkXLl_yokfS4,2075
@@ -23,9 +23,9 @@ tmnt/utils/pubmed_utils.py,sha256=3sHwoun7vxb0GV-arhpXLMUbAZne0huAh9xQNy6H40E,12
 tmnt/utils/random.py,sha256=qY75WG3peWoMh9pUyCPBEo6q8IvkF6VRjeb5CqJOBF8,327
 tmnt/utils/recalibrate.py,sha256=TmpB8An8bslICZ13UTJfIvr8VoqiSedtpHxec4n8CHk,1439
 tmnt/utils/vocab.py,sha256=J6GFGLyvDgdmtVQjYlyzWjuykRD3kllCKPG1z0lI0P8,3504
-tmnt-0.7.60.dist-info/licenses/LICENSE,sha256=qFZJrfJ7Zi4IXDiyiGVrHWic_l1h2tc36tI8Z7rK9bs,11356
-tmnt-0.7.60.dist-info/licenses/NOTICE,sha256=p0kYIVAkReTFaGb4C-qPa7h5ztze6hGzOpjCMMbOipU,425
-tmnt-0.7.60.dist-info/METADATA,sha256=DbmpuasEyoW6FNHmu-YfNsgq5J1lV0MjWPe04CCq8Fk,1663
-tmnt-0.7.60.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-tmnt-0.7.60.dist-info/top_level.txt,sha256=RpYgUl187sXnqmiwKjZZdcDlHz2AALs6bGdUcukyd_E,5
-tmnt-0.7.60.dist-info/RECORD,,
+tmnt-0.7.61.dist-info/licenses/LICENSE,sha256=qFZJrfJ7Zi4IXDiyiGVrHWic_l1h2tc36tI8Z7rK9bs,11356
+tmnt-0.7.61.dist-info/licenses/NOTICE,sha256=p0kYIVAkReTFaGb4C-qPa7h5ztze6hGzOpjCMMbOipU,425
+tmnt-0.7.61.dist-info/METADATA,sha256=hyptC3YXersP-7oR89NgKU4UksTrvsbcC5Ml38WSYG0,1663
+tmnt-0.7.61.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+tmnt-0.7.61.dist-info/top_level.txt,sha256=RpYgUl187sXnqmiwKjZZdcDlHz2AALs6bGdUcukyd_E,5
+tmnt-0.7.61.dist-info/RECORD,,