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.

Files changed (31) hide show
  1. {sure_tools-2.3.1 → sure_tools-2.3.2}/PKG-INFO +1 -1
  2. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/DensityFlow.py +11 -4
  3. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/PKG-INFO +1 -1
  4. {sure_tools-2.3.1 → sure_tools-2.3.2}/setup.py +1 -1
  5. {sure_tools-2.3.1 → sure_tools-2.3.2}/LICENSE +0 -0
  6. {sure_tools-2.3.1 → sure_tools-2.3.2}/README.md +0 -0
  7. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/PerturbE.py +0 -0
  8. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/SURE.py +0 -0
  9. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/__init__.py +0 -0
  10. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/assembly/__init__.py +0 -0
  11. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/assembly/assembly.py +0 -0
  12. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/assembly/atlas.py +0 -0
  13. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/atac/__init__.py +0 -0
  14. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/atac/utils.py +0 -0
  15. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/codebook/__init__.py +0 -0
  16. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/codebook/codebook.py +0 -0
  17. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/flow/__init__.py +0 -0
  18. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/flow/flow_stats.py +0 -0
  19. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/flow/plot_quiver.py +0 -0
  20. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/perturb/__init__.py +0 -0
  21. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/perturb/perturb.py +0 -0
  22. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/utils/__init__.py +0 -0
  23. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/utils/custom_mlp.py +0 -0
  24. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/utils/queue.py +0 -0
  25. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE/utils/utils.py +0 -0
  26. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/SOURCES.txt +0 -0
  27. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/dependency_links.txt +0 -0
  28. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/entry_points.txt +0 -0
  29. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/requires.txt +0 -0
  30. {sure_tools-2.3.1 → sure_tools-2.3.2}/SURE_tools.egg-info/top_level.txt +0 -0
  31. {sure_tools-2.3.1 → sure_tools-2.3.2}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.3.1
3
+ Version: 2.3.2
4
4
  Summary: Succinct Representation of Single Cells
5
5
  Home-page: https://github.com/ZengFLab/SURE
6
6
  Author: Feng Zeng
@@ -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
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.3.1
3
+ Version: 2.3.2
4
4
  Summary: Succinct Representation of Single Cells
5
5
  Home-page: https://github.com/ZengFLab/SURE
6
6
  Author: Feng Zeng
@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
5
5
 
6
6
  setup(
7
7
  name='SURE-tools',
8
- version='2.3.1',
8
+ version='2.3.2',
9
9
  description='Succinct Representation of Single Cells',
10
10
  long_description=long_description,
11
11
  long_description_content_type="text/markdown",
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes