SURE-tools 2.3.1__tar.gz → 2.3.2__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.3.1 → sure_tools-2.3.2}/PKG-INFO +1 -1
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/DensityFlow.py +11 -4
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/PKG-INFO +1 -1
- {sure_tools-2.3.1 → sure_tools-2.3.2}/setup.py +1 -1
- {sure_tools-2.3.1 → sure_tools-2.3.2}/LICENSE +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/README.md +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/PerturbE.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/SURE.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/__init__.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/assembly/__init__.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/assembly/assembly.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/assembly/atlas.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/atac/__init__.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/atac/utils.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/codebook/__init__.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/codebook/codebook.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/flow/__init__.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/flow/flow_stats.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/flow/plot_quiver.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/perturb/__init__.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/perturb/perturb.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/utils/__init__.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/utils/custom_mlp.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/utils/queue.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/utils/utils.py +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/SOURCES.txt +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/dependency_links.txt +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/entry_points.txt +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/requires.txt +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/top_level.txt +0 -0
- {sure_tools-2.3.1 → sure_tools-2.3.2}/setup.cfg +0 -0
|
@@ -396,7 +396,8 @@ class DensityFlow(nn.Module):
|
|
|
396
396
|
else:
|
|
397
397
|
pyro.sample('x', dist.Poisson(rate=rate).to_event(1), obs=xs.round())
|
|
398
398
|
elif self.loss_func == 'multinomial':
|
|
399
|
-
pyro.sample('x', dist.Multinomial(total_count=int(1e8), probs=theta), obs=xs)
|
|
399
|
+
#pyro.sample('x', dist.Multinomial(total_count=int(1e8), probs=theta), obs=xs)
|
|
400
|
+
pyro.sample('x', dist.Multinomial(total_count=int(1e8), logits=concentrate), obs=xs)
|
|
400
401
|
elif self.loss_func == 'bernoulli':
|
|
401
402
|
if self.use_zeroinflate:
|
|
402
403
|
pyro.sample('x', dist.ZeroInflatedDistribution(dist.Bernoulli(logits=log_theta),gate_logits=gate_logits).to_event(1), obs=xs)
|
|
@@ -480,7 +481,8 @@ class DensityFlow(nn.Module):
|
|
|
480
481
|
else:
|
|
481
482
|
pyro.sample('x', dist.Poisson(rate=rate).to_event(1), obs=xs.round())
|
|
482
483
|
elif self.loss_func == 'multinomial':
|
|
483
|
-
pyro.sample('x', dist.Multinomial(total_count=int(1e8), probs=theta), obs=xs)
|
|
484
|
+
#pyro.sample('x', dist.Multinomial(total_count=int(1e8), probs=theta), obs=xs)
|
|
485
|
+
pyro.sample('x', dist.Multinomial(total_count=int(1e8), logits=concentrate), obs=xs)
|
|
484
486
|
elif self.loss_func == 'bernoulli':
|
|
485
487
|
if self.use_zeroinflate:
|
|
486
488
|
pyro.sample('x', dist.ZeroInflatedDistribution(dist.Bernoulli(logits=log_theta),gate_logits=gate_logits).to_event(1), obs=xs)
|
|
@@ -576,7 +578,8 @@ class DensityFlow(nn.Module):
|
|
|
576
578
|
else:
|
|
577
579
|
pyro.sample('x', dist.Poisson(rate=rate).to_event(1), obs=xs.round())
|
|
578
580
|
elif self.loss_func == 'multinomial':
|
|
579
|
-
pyro.sample('x', dist.Multinomial(total_count=int(1e8), probs=theta), obs=xs)
|
|
581
|
+
#pyro.sample('x', dist.Multinomial(total_count=int(1e8), probs=theta), obs=xs)
|
|
582
|
+
pyro.sample('x', dist.Multinomial(total_count=int(1e8), logits=concentrate), obs=xs)
|
|
580
583
|
elif self.loss_func == 'bernoulli':
|
|
581
584
|
if self.use_zeroinflate:
|
|
582
585
|
pyro.sample('x', dist.ZeroInflatedDistribution(dist.Bernoulli(logits=log_theta),gate_logits=gate_logits).to_event(1), obs=xs)
|
|
@@ -682,7 +685,8 @@ class DensityFlow(nn.Module):
|
|
|
682
685
|
else:
|
|
683
686
|
pyro.sample('x', dist.Poisson(rate=rate).to_event(1), obs=xs.round())
|
|
684
687
|
elif self.loss_func == 'multinomial':
|
|
685
|
-
pyro.sample('x', dist.Multinomial(total_count=int(1e8), probs=theta), obs=xs)
|
|
688
|
+
#pyro.sample('x', dist.Multinomial(total_count=int(1e8), probs=theta), obs=xs)
|
|
689
|
+
pyro.sample('x', dist.Multinomial(total_count=int(1e8), logits=concentrate), obs=xs)
|
|
686
690
|
elif self.loss_func == 'bernoulli':
|
|
687
691
|
if self.use_zeroinflate:
|
|
688
692
|
pyro.sample('x', dist.ZeroInflatedDistribution(dist.Bernoulli(logits=log_theta),gate_logits=gate_logits).to_event(1), obs=xs)
|
|
@@ -946,6 +950,9 @@ class DensityFlow(nn.Module):
|
|
|
946
950
|
if self.loss_func == 'bernoulli':
|
|
947
951
|
#counts = self.sigmoid(concentrate)
|
|
948
952
|
counts = dist.Bernoulli(logits=concentrate).to_event(1).mean
|
|
953
|
+
elif self.loss_func == 'multinomial':
|
|
954
|
+
theta = dist.Multinomial(total_count=int(1e8), logits=concentrate).mean
|
|
955
|
+
counts = theta * library_size
|
|
949
956
|
else:
|
|
950
957
|
rate = concentrate.exp()
|
|
951
958
|
theta = dist.DirichletMultinomial(total_count=1, concentration=rate).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
|
|
File without changes
|