SURE-tools 2.1.91__tar.gz → 2.1.92__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.
- {sure_tools-2.1.91 → sure_tools-2.1.92}/PKG-INFO +1 -1
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/PerturbFlow.py +2 -10
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE_tools.egg-info/PKG-INFO +1 -1
- {sure_tools-2.1.91 → sure_tools-2.1.92}/setup.py +1 -1
- {sure_tools-2.1.91 → sure_tools-2.1.92}/LICENSE +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/README.md +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/SURE.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/__init__.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/assembly/__init__.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/assembly/assembly.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/assembly/atlas.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/atac/__init__.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/atac/utils.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/codebook/__init__.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/codebook/codebook.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/flow/__init__.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/flow/flow_stats.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/flow/plot_quiver.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/perturb/__init__.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/perturb/perturb.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/utils/__init__.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/utils/custom_mlp.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/utils/queue.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE/utils/utils.py +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE_tools.egg-info/SOURCES.txt +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE_tools.egg-info/dependency_links.txt +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE_tools.egg-info/entry_points.txt +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE_tools.egg-info/requires.txt +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/SURE_tools.egg-info/top_level.txt +0 -0
- {sure_tools-2.1.91 → sure_tools-2.1.92}/setup.cfg +0 -0
|
@@ -59,7 +59,6 @@ class PerturbFlow(nn.Module):
|
|
|
59
59
|
input_size: int,
|
|
60
60
|
codebook_size: int = 200,
|
|
61
61
|
cell_factor_size: int = 0,
|
|
62
|
-
cell_factor_effect_discrete: bool = False,
|
|
63
62
|
supervised_mode: bool = False,
|
|
64
63
|
z_dim: int = 10,
|
|
65
64
|
z_dist: Literal['normal','studentt','laplacian','cauchy','gumbel'] = 'gumbel',
|
|
@@ -103,7 +102,6 @@ class PerturbFlow(nn.Module):
|
|
|
103
102
|
else:
|
|
104
103
|
self.use_bias = [not zero_bias] * self.cell_factor_size
|
|
105
104
|
#self.use_bias = not zero_bias
|
|
106
|
-
self.enumrate = cell_factor_effect_discrete
|
|
107
105
|
|
|
108
106
|
self.codebook_weights = None
|
|
109
107
|
|
|
@@ -431,10 +429,7 @@ class PerturbFlow(nn.Module):
|
|
|
431
429
|
zns = pyro.sample('zn', dist.Gumbel(zn_loc, zn_scale).to_event(1))
|
|
432
430
|
|
|
433
431
|
if self.cell_factor_size>0:
|
|
434
|
-
|
|
435
|
-
zus = self._total_effects(zn_loc, us)
|
|
436
|
-
else:
|
|
437
|
-
zus = self._total_effects(zns, us)
|
|
432
|
+
zus = self._total_effects(zns, us)
|
|
438
433
|
zs = zns+zus
|
|
439
434
|
else:
|
|
440
435
|
zs = zns
|
|
@@ -636,10 +631,7 @@ class PerturbFlow(nn.Module):
|
|
|
636
631
|
# zus = self.cell_factor_effect[i]([zns,us[:,i].reshape(-1,1)])
|
|
637
632
|
# else:
|
|
638
633
|
# zus = zus + self.cell_factor_effect[i]([zns,us[:,i].reshape(-1,1)])
|
|
639
|
-
|
|
640
|
-
zus = self._total_effects(zn_loc, us)
|
|
641
|
-
else:
|
|
642
|
-
zus = self._total_effects(zns, us)
|
|
634
|
+
zus = self._total_effects(zns, us)
|
|
643
635
|
zs = zns+zus
|
|
644
636
|
else:
|
|
645
637
|
zs = zns
|
|
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
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|