SURE-tools 2.1.46__py3-none-any.whl → 2.1.48__py3-none-any.whl

SURE/PerturbFlow.py

@@ -378,7 +378,8 @@ class PerturbFlow(nn.Module):
 
     def guide1(self, xs):
         with pyro.plate('data'):
-            zn_loc, zn_scale = self.encoder_zn(xs)
+            #zn_loc, zn_scale = self.encoder_zn(xs)
+            zn_loc, zn_scale = self._get_basal_embedding(xs)
             zns = pyro.sample('zn', dist.Normal(zn_loc, zn_scale).to_event(1))
 
             alpha = self.encoder_n(zns)
@@ -466,7 +467,8 @@ class PerturbFlow(nn.Module):
 
     def guide2(self, xs, us=None):
         with pyro.plate('data'):
-            zn_loc, zn_scale = self.encoder_zn(xs)
+            #zn_loc, zn_scale = self.encoder_zn(xs)
+            zn_loc, zn_scale = self._get_basal_embedding(xs)
             zns = pyro.sample('zn', dist.Normal(zn_loc, zn_scale).to_event(1))
 
             alpha = self.encoder_n(zns)
@@ -561,7 +563,8 @@ class PerturbFlow(nn.Module):
     def guide3(self, xs, ys, embeds=None):
         with pyro.plate('data'):
             if embeds is None:
-                zn_loc, zn_scale = self.encoder_zn(xs)
+                #zn_loc, zn_scale = self.encoder_zn(xs)
+                zn_loc, zn_scale = self._get_basal_embedding(xs)
                 zns = pyro.sample('zn', dist.Normal(zn_loc, zn_scale).to_event(1))
 
     def model4(self, xs, us, ys, embeds=None):
@@ -663,7 +666,8 @@ class PerturbFlow(nn.Module):
     def guide4(self, xs, us, ys, embeds=None):
         with pyro.plate('data'):
             if embeds is None:
-                zn_loc, zn_scale = self.encoder_zn(xs)
+                #zn_loc, zn_scale = self.encoder_zn(xs)
+                zn_loc, zn_scale = self._get_basal_embedding(xs)
                 zns = pyro.sample('zn', dist.Normal(zn_loc, zn_scale).to_event(1))
 
     def _total_effects(self, zns, us):
@@ -692,8 +696,8 @@ class PerturbFlow(nn.Module):
         return cb
 
     def _get_basal_embedding(self, xs):           
-        zns, _ = self.encoder_zn(xs)
-        return zns 
+        loc, scale = self.encoder_zn(xs)
+        return loc, scale 
     
     def get_basal_embedding(self, 
                              xs, 
@@ -720,7 +724,7 @@ class PerturbFlow(nn.Module):
         Z = []
         with tqdm(total=len(dataloader), desc='', unit='batch') as pbar:
             for X_batch, _ in dataloader:
-                zns = self._get_basal_embedding(X_batch)
+                zns,_ = self._get_basal_embedding(X_batch)
                 Z.append(tensor_to_numpy(zns))
                 pbar.update(1)
 
@@ -732,7 +736,7 @@ class PerturbFlow(nn.Module):
             alpha = self.encoder_n(xs)
         else:
             #zns,_ = self.encoder_zn(xs)
-            zns = self._get_basal_embedding(xs)
+            zns,_ = self._get_basal_embedding(xs)
             alpha = self.encoder_n(zns)
         return alpha
     
@@ -803,7 +807,7 @@ class PerturbFlow(nn.Module):
     
     def _cell_response(self, xs, factor_idx, perturb):
         #zns,_ = self.encoder_zn(xs)    
-        zns = self._get_basal_embedding(xs)   
+        zns,_ = self._get_basal_embedding(xs)   
         if perturb.ndim==2:
             ms = self.cell_factor_effect[factor_idx]([zns, perturb])
         else:

SURE/perturb/perturb.py

@@ -9,7 +9,7 @@ class LabelMatrix:
     def __init__(self):
         self.labels_ = None
     
-    def fit_transform(self, labels, control_label=None, sep_pattern=r'[;_\-\s]', speedup: Literal['none','vectorize','parallel']='none'):
+    def fit_transform(self, labels, control_label=None, sep_pattern=r'[,;_\s]', speedup: Literal['none','vectorize','parallel']='none'):
         if speedup=='none':
             mat, self.labels_ = label_to_matrix(labels=labels, sep_pattern=sep_pattern)
         elif speedup=='vectorize':

SURE/utils/__init__.py

@@ -7,7 +7,7 @@ from .utils import find_partitions_greedy
 
 from .queue import PriorityQueue
 
-from .custom_mlp import MLP, Exp, ZeroBiasMLP
+from .custom_mlp import MLP, Exp, ZeroBiasMLP, HDMLP
 
 # Importing modules
 #from . import utils

SURE/utils/custom_mlp.py

@@ -241,4 +241,38 @@ class ZeroBiasMLP(nn.Module):
         mask = torch.zeros_like(y)
         mask[x[1][:,0]>0,:] = 1
         return y*mask
-        
+        
+
+class HDMLP(nn.Module):
+    def __init__(
+        self,
+        input_size,
+        hidden_sizes,
+        output_depth,
+        activation=nn.ReLU,
+        output_activation=None,
+        post_layer_fct=lambda layer_ix, total_layers, layer: None,
+        post_act_fct=lambda layer_ix, total_layers, layer: None,
+        allow_broadcast=False,
+        use_cuda=False,
+    ):
+        # init the module object
+        super().__init__()
+        self.mlp = MLP(mlp_sizes=[1] + hidden_sizes + [output_depth],
+                       activation=activation,
+                       output_activation=output_activation,
+                       post_layer_fct=post_layer_fct,
+                       post_act_fct=post_act_fct,
+                       allow_broadcast=allow_broadcast,
+                       use_cuda=use_cuda,
+                       bias=True)
+        self.input_size=input_size
+        self.output_depth=output_depth
+        
+    # pass through our sequential for the output!
+    def forward(self, x):
+        batch_size, n = x.shape
+        x = x.view(batch_size * n, 1)
+        out = self.mlp(x)
+        out = out.view(batch_size, n, self.output_depth)
+        return out

{sure_tools-2.1.46.dist-info → sure_tools-2.1.48.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SURE-tools
-Version: 2.1.46
+Version: 2.1.48
 Summary: Succinct Representation of Single Cells
 Home-page: https://github.com/ZengFLab/SURE
 Author: Feng Zeng

{sure_tools-2.1.46.dist-info → sure_tools-2.1.48.dist-info}/RECORD

@@ -1,4 +1,4 @@
-SURE/PerturbFlow.py,sha256=EjBvHUNmQeyF41H-rzg7fSrcKo9_ts2za-JiMRp4y9M,52394
+SURE/PerturbFlow.py,sha256=0Zalme94qlUBnom7lY3OmUZcyLPd2F69QJ29LjuCpd8,52667
 SURE/SURE.py,sha256=ko15a9BhvUqHviogZ0YCdTQjM-2zqkO9OvHZSpnGbg0,47458
 SURE/__init__.py,sha256=NOJI_K-eCqPgStXXvgl3wIEMp6d8saMTDYLJ7Ga9MqE,293
 SURE/assembly/__init__.py,sha256=jxZLURXKPzXe21LhrZ09LgZr33iqdjlQy4oSEj5gR2Q,172
@@ -12,14 +12,14 @@ SURE/flow/__init__.py,sha256=rsAjYsh1xVIrxBCuwOE0Q_6N5th1wBgjJceV0ABPG3c,183
 SURE/flow/flow_stats.py,sha256=_pF7m4-87SKlCHVtVmx3LG2bAGVXOnAfEgMzLhLx4Io,10910
 SURE/flow/plot_quiver.py,sha256=UbmuScUcgbQHeMmjKmgqxjrIjHhiHx0VWct16UMMwuE,8110
 SURE/perturb/__init__.py,sha256=8TP1dSUhXiZzKpFebHZmm8XMMGbUz_OfQ10xu-6uPPY,43
-SURE/perturb/perturb.py,sha256=ICs-yMIuT0tZtyx9R7tkzzhAcurs_eActv2v3uyjreE,5038
-SURE/utils/__init__.py,sha256=QJUOfrXzdWSmoM0P3LH8oKEHttzCWqpDy2UF0F0dtN4,673
-SURE/utils/custom_mlp.py,sha256=rHnx9jEef02zfCUdbYVCmbuHcDdIBmRgt__wpdpZvYg,8104
+SURE/perturb/perturb.py,sha256=1iSsCePcwkA2CyM1nCdq_G8gogUNjhMH0BfhhvhpJQk,5037
+SURE/utils/__init__.py,sha256=YF5jB-PAHJQ40OlcZ7BCZbsN2q1JKuPT6EppilRXQqM,680
+SURE/utils/custom_mlp.py,sha256=C0EXLGYsWkUQpEL49AyBFPSzKmasb2hdvtnJfxbF-YU,9282
 SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
 SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
-sure_tools-2.1.46.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
-sure_tools-2.1.46.dist-info/METADATA,sha256=bx8NRcUtUXrKsj13OhmWzb49MxtVxI9KdrF_BEAglMc,2651
-sure_tools-2.1.46.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sure_tools-2.1.46.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
-sure_tools-2.1.46.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
-sure_tools-2.1.46.dist-info/RECORD,,
+sure_tools-2.1.48.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
+sure_tools-2.1.48.dist-info/METADATA,sha256=sjCCzYJQAUe3MpF9a4fsEoXBRhs83VC5ez5ar4z-cX4,2651
+sure_tools-2.1.48.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sure_tools-2.1.48.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
+sure_tools-2.1.48.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
+sure_tools-2.1.48.dist-info/RECORD,,