SURE-tools 2.0.9__py3-none-any.whl → 2.0.10__py3-none-any.whl

SURE/SURE.py

@@ -860,6 +860,30 @@ class SURE(nn.Module):
         ms = self.cell_factor_effect[factor_idx]([zs,ps])
         return tensor_to_numpy(ms)
     
+    def _get_expression_responses(self, delta_zs):
+        return self.decoder_concentrate(delta_zs)
+    
+    def get_expression_responses(self, 
+                             delta_zs, 
+                             batch_size: int = 1024):
+        """
+        Return cells' changes in the latent space induced by specific perturbation of a factor
+
+        """
+        delta_zs = convert_to_tensor(delta_zs, device=self.get_device())
+        dataset = CustomDataset(delta_zs)
+        dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=False)
+
+        R = []
+        with tqdm(total=len(dataloader), desc='', unit='batch') as pbar:
+            for delta_Z_batch, _ in dataloader:
+                r = self._cell_move(delta_Z_batch)
+                R.append(tensor_to_numpy(r))
+                pbar.update(1)
+
+        R = np.concatenate(R)
+        return R
+    
     def preprocess(self, xs, threshold=0):
         if self.loss_func == 'bernoulli':
             ad = sc.AnnData(xs)

{sure_tools-2.0.9.dist-info → sure_tools-2.0.10.dist-info}/METADATA

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

{sure_tools-2.0.9.dist-info → sure_tools-2.0.10.dist-info}/RECORD

@@ -1,4 +1,4 @@
-SURE/SURE.py,sha256=WSRDPU-iICy0nyBF2aGGcQiJDzdlbLjsRvVBaTXRb7k,51004
+SURE/SURE.py,sha256=DzuRXp9AtnwoezsR-zTkm6PxlfOV8sdmu3mYwkapYQs,51875
 SURE/SURE2.py,sha256=8wlnMwb1xuf9QUksNkWdWx5ZWq-xIy9NLx8RdUnE82o,48501
 SURE/__init__.py,sha256=mxdLoWa2fPEM4-WkJetMfcecpzqJ8YsM4lOoKHSLTrU,207
 SURE/assembly/__init__.py,sha256=jxZLURXKPzXe21LhrZ09LgZr33iqdjlQy4oSEj5gR2Q,172
@@ -15,9 +15,9 @@ SURE/utils/__init__.py,sha256=Htqv4KqVKcRiaaTBsR-6yZ4LSlbhbzutjNKXGD9-uds,660
 SURE/utils/custom_mlp.py,sha256=07TYX1HgxfEjb_3i5MpiZfNhOhx3dKntuwGkrpteWiM,7036
 SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
 SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
-sure_tools-2.0.9.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
-sure_tools-2.0.9.dist-info/METADATA,sha256=5sRH5YJRo1b-gUxMOIIFrbcLuRmGIwV8fPqwT8emFUs,2650
-sure_tools-2.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sure_tools-2.0.9.dist-info/entry_points.txt,sha256=u12payZYgCBy5FCwRHP6AlSQhKCiWSEDwj68r1DVdn8,40
-sure_tools-2.0.9.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
-sure_tools-2.0.9.dist-info/RECORD,,
+sure_tools-2.0.10.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
+sure_tools-2.0.10.dist-info/METADATA,sha256=_64vkwUxRdYs5xvh76oYB34PUDpEnCN5Inl5tlGhe38,2651
+sure_tools-2.0.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sure_tools-2.0.10.dist-info/entry_points.txt,sha256=u12payZYgCBy5FCwRHP6AlSQhKCiWSEDwj68r1DVdn8,40
+sure_tools-2.0.10.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
+sure_tools-2.0.10.dist-info/RECORD,,