SURE-tools 2.0.6__tar.gz → 2.0.8__tar.gz
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.
Potentially problematic release.
This version of SURE-tools might be problematic. Click here for more details.
- {sure_tools-2.0.6 → sure_tools-2.0.8}/PKG-INFO +1 -1
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/SURE.py +11 -1
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/flow/plot_quiver.py +1 -1
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE_tools.egg-info/PKG-INFO +1 -1
- {sure_tools-2.0.6 → sure_tools-2.0.8}/setup.py +1 -1
- {sure_tools-2.0.6 → sure_tools-2.0.8}/LICENSE +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/README.md +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/__init__.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/assembly/__init__.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/assembly/assembly.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/assembly/atlas.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/atac/__init__.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/atac/utils.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/codebook/__init__.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/codebook/codebook.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/flow/__init__.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/utils/__init__.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/utils/custom_mlp.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/utils/queue.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE/utils/utils.py +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE_tools.egg-info/SOURCES.txt +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE_tools.egg-info/dependency_links.txt +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE_tools.egg-info/entry_points.txt +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE_tools.egg-info/requires.txt +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/SURE_tools.egg-info/top_level.txt +0 -0
- {sure_tools-2.0.6 → sure_tools-2.0.8}/setup.cfg +0 -0
|
@@ -814,7 +814,7 @@ class SURE(nn.Module):
|
|
|
814
814
|
return A
|
|
815
815
|
|
|
816
816
|
def _cell_move(self, xs, factor_idx, perturb):
|
|
817
|
-
zns = self.encoder_zn(xs)
|
|
817
|
+
zns,_ = self.encoder_zn(xs)
|
|
818
818
|
if type(factor_idx) == str:
|
|
819
819
|
factor_idx = np.where(self.cell_factor_names==factor_idx)
|
|
820
820
|
|
|
@@ -850,6 +850,16 @@ class SURE(nn.Module):
|
|
|
850
850
|
Z = np.concatenate(Z)
|
|
851
851
|
return Z
|
|
852
852
|
|
|
853
|
+
def get_metacell_move(self, factor_idx, perturb):
|
|
854
|
+
zs = self._get_codebook()
|
|
855
|
+
ps = convert_to_tensor(perturb, device=self.get_device())
|
|
856
|
+
|
|
857
|
+
if type(factor_idx) == str:
|
|
858
|
+
factor_idx = np.where(self.cell_factor_names==factor_idx)
|
|
859
|
+
|
|
860
|
+
ms = self.cell_factor_effect[factor_idx]([zs,ps])
|
|
861
|
+
return tensor_to_numpy(ms)
|
|
862
|
+
|
|
853
863
|
def preprocess(self, xs, threshold=0):
|
|
854
864
|
if self.loss_func == 'bernoulli':
|
|
855
865
|
ad = sc.AnnData(xs)
|
|
@@ -32,7 +32,7 @@ def plot_quiver(z_points, delta_z, method='umap', figsize=(6,4), dpi=200):
|
|
|
32
32
|
sc.pp.neighbors(ad)
|
|
33
33
|
sc.tl.umap(ad)
|
|
34
34
|
z_2d = ad[:z_points.shape[0]].obsm['X_umap']
|
|
35
|
-
delta_z_2d = ad[z_points.shape[0]:] - z_2d
|
|
35
|
+
delta_z_2d = ad[z_points.shape[0]:].obsm['X_umap'] - z_2d
|
|
36
36
|
dim_names = ['UMAP1', 'UMAP2']
|
|
37
37
|
|
|
38
38
|
# 绘制quiver图
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|