SURE-tools 2.1.73__tar.gz → 2.1.74__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.1.73 → sure_tools-2.1.74}/PKG-INFO +1 -1
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/PerturbFlow.py +7 -1
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE_tools.egg-info/PKG-INFO +1 -1
- {sure_tools-2.1.73 → sure_tools-2.1.74}/setup.py +1 -1
- {sure_tools-2.1.73 → sure_tools-2.1.74}/LICENSE +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/README.md +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/SURE.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/__init__.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/assembly/__init__.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/assembly/assembly.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/assembly/atlas.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/atac/__init__.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/atac/utils.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/codebook/__init__.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/codebook/codebook.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/flow/__init__.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/flow/flow_stats.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/flow/plot_quiver.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/perturb/__init__.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/perturb/perturb.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/utils/__init__.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/utils/custom_mlp.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/utils/queue.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE/utils/utils.py +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE_tools.egg-info/SOURCES.txt +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE_tools.egg-info/dependency_links.txt +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE_tools.egg-info/entry_points.txt +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE_tools.egg-info/requires.txt +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/SURE_tools.egg-info/top_level.txt +0 -0
- {sure_tools-2.1.73 → sure_tools-2.1.74}/setup.cfg +0 -0
|
@@ -104,6 +104,7 @@ class PerturbFlow(nn.Module):
|
|
|
104
104
|
#self.use_bias = not zero_bias
|
|
105
105
|
|
|
106
106
|
self.codebook_weights = None
|
|
107
|
+
self.total_count = None
|
|
107
108
|
|
|
108
109
|
set_random_seed(seed)
|
|
109
110
|
self.setup_networks()
|
|
@@ -317,6 +318,7 @@ class PerturbFlow(nn.Module):
|
|
|
317
318
|
if self.loss_func=='negbinomial':
|
|
318
319
|
total_count = pyro.param("inverse_dispersion", self.inverse_dispersion *
|
|
319
320
|
xs.new_ones(self.input_size), constraint=constraints.positive)
|
|
321
|
+
self.total_count = total_count
|
|
320
322
|
|
|
321
323
|
if self.use_zeroinflate:
|
|
322
324
|
gate_logits = pyro.param("dropout_rate", xs.new_zeros(self.input_size))
|
|
@@ -396,6 +398,7 @@ class PerturbFlow(nn.Module):
|
|
|
396
398
|
if self.loss_func=='negbinomial':
|
|
397
399
|
total_count = pyro.param("inverse_dispersion", self.inverse_dispersion *
|
|
398
400
|
xs.new_ones(self.input_size), constraint=constraints.positive)
|
|
401
|
+
self.total_count = total_count
|
|
399
402
|
|
|
400
403
|
if self.use_zeroinflate:
|
|
401
404
|
gate_logits = pyro.param("dropout_rate", xs.new_zeros(self.input_size))
|
|
@@ -480,6 +483,7 @@ class PerturbFlow(nn.Module):
|
|
|
480
483
|
if self.loss_func=='negbinomial':
|
|
481
484
|
total_count = pyro.param("inverse_dispersion", self.inverse_dispersion *
|
|
482
485
|
xs.new_ones(self.input_size), constraint=constraints.positive)
|
|
486
|
+
self.total_count = total_count
|
|
483
487
|
|
|
484
488
|
if self.use_zeroinflate:
|
|
485
489
|
gate_logits = pyro.param("dropout_rate", xs.new_zeros(self.input_size))
|
|
@@ -576,6 +580,7 @@ class PerturbFlow(nn.Module):
|
|
|
576
580
|
if self.loss_func=='negbinomial':
|
|
577
581
|
total_count = pyro.param("inverse_dispersion", self.inverse_dispersion *
|
|
578
582
|
xs.new_ones(self.input_size), constraint=constraints.positive)
|
|
583
|
+
self.total_count = total_count
|
|
579
584
|
|
|
580
585
|
if self.use_zeroinflate:
|
|
581
586
|
gate_logits = pyro.param("dropout_rate", xs.new_zeros(self.input_size))
|
|
@@ -880,7 +885,8 @@ class PerturbFlow(nn.Module):
|
|
|
880
885
|
rate = concentrate.exp()
|
|
881
886
|
theta = dist.DirichletMultinomial(total_count=1, concentration=rate).mean
|
|
882
887
|
|
|
883
|
-
total_count =
|
|
888
|
+
total_count = self.total_count
|
|
889
|
+
#total_count = pyro.param("inverse_dispersion")
|
|
884
890
|
#store = pyro.get_param_store()
|
|
885
891
|
#total_count = store['inverse_dispersion']
|
|
886
892
|
counts = dist.NegativeBinomial(total_count=total_count, probs=theta).to_event(1).mean
|
|
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
|