SURE-tools 2.2.1__tar.gz → 2.2.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 (30) hide show
  1. {sure_tools-2.2.1 → sure_tools-2.2.2}/PKG-INFO +1 -1
  2. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/DensityFlow.py +2 -2
  3. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/SURE.py +6 -6
  4. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE_tools.egg-info/PKG-INFO +1 -1
  5. {sure_tools-2.2.1 → sure_tools-2.2.2}/setup.py +1 -1
  6. {sure_tools-2.2.1 → sure_tools-2.2.2}/LICENSE +0 -0
  7. {sure_tools-2.2.1 → sure_tools-2.2.2}/README.md +0 -0
  8. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/__init__.py +0 -0
  9. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/assembly/__init__.py +0 -0
  10. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/assembly/assembly.py +0 -0
  11. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/assembly/atlas.py +0 -0
  12. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/atac/__init__.py +0 -0
  13. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/atac/utils.py +0 -0
  14. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/codebook/__init__.py +0 -0
  15. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/codebook/codebook.py +0 -0
  16. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/flow/__init__.py +0 -0
  17. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/flow/flow_stats.py +0 -0
  18. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/flow/plot_quiver.py +0 -0
  19. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/perturb/__init__.py +0 -0
  20. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/perturb/perturb.py +0 -0
  21. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/utils/__init__.py +0 -0
  22. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/utils/custom_mlp.py +0 -0
  23. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/utils/queue.py +0 -0
  24. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE/utils/utils.py +0 -0
  25. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE_tools.egg-info/SOURCES.txt +0 -0
  26. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE_tools.egg-info/dependency_links.txt +0 -0
  27. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE_tools.egg-info/entry_points.txt +0 -0
  28. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE_tools.egg-info/requires.txt +0 -0
  29. {sure_tools-2.2.1 → sure_tools-2.2.2}/SURE_tools.egg-info/top_level.txt +0 -0
  30. {sure_tools-2.2.1 → sure_tools-2.2.2}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.2.1
3
+ Version: 2.2.2
4
4
  Summary: Succinct Representation of Single Cells
5
5
  Home-page: https://github.com/ZengFLab/SURE
6
6
  Author: Feng Zeng
@@ -64,8 +64,8 @@ class DensityFlow(nn.Module):
64
64
  z_dist: Literal['normal','studentt','laplacian','cauchy','gumbel'] = 'gumbel',
65
65
  loss_func: Literal['negbinomial','poisson','multinomial','bernoulli'] = 'poisson',
66
66
  inverse_dispersion: float = 10.0,
67
- use_zeroinflate: bool = False,
68
- hidden_layers: list = [300],
67
+ use_zeroinflate: bool = True,
68
+ hidden_layers: list = [500],
69
69
  hidden_layer_activation: Literal['relu','softplus','leakyrelu','linear'] = 'relu',
70
70
  nn_dropout: float = 0.1,
71
71
  post_layer_fct: list = ['layernorm'],
@@ -99,17 +99,17 @@ class SURE(nn.Module):
99
99
  cell_factor_size: int = 0,
100
100
  supervised_mode: bool = False,
101
101
  z_dim: int = 10,
102
- z_dist: Literal['normal','studentt','laplacian','cauchy','gumbel'] = 'normal',
103
- loss_func: Literal['negbinomial','poisson','multinomial','bernoulli'] = 'negbinomial',
102
+ z_dist: Literal['normal','studentt','laplacian','cauchy','gumbel'] = 'gumbel',
103
+ loss_func: Literal['negbinomial','poisson','multinomial','bernoulli'] = 'poisson',
104
104
  inverse_dispersion: float = 10.0,
105
105
  use_zeroinflate: bool = True,
106
- hidden_layers: list = [300],
106
+ hidden_layers: list = [500],
107
107
  hidden_layer_activation: Literal['relu','softplus','leakyrelu','linear'] = 'relu',
108
108
  nn_dropout: float = 0.1,
109
109
  post_layer_fct: list = ['layernorm'],
110
110
  post_act_fct: list = None,
111
111
  config_enum: str = 'parallel',
112
- use_cuda: bool = False,
112
+ use_cuda: bool = True,
113
113
  seed: int = 42,
114
114
  dtype = torch.float32, # type: ignore
115
115
  ):
@@ -817,7 +817,7 @@ class SURE(nn.Module):
817
817
  us = None,
818
818
  ys = None,
819
819
  zs = None,
820
- num_epochs: int = 200,
820
+ num_epochs: int = 500,
821
821
  learning_rate: float = 0.0001,
822
822
  batch_size: int = 256,
823
823
  algo: Literal['adam','rmsprop','adamw'] = 'adam',
@@ -826,7 +826,7 @@ class SURE(nn.Module):
826
826
  decay_rate: float = 0.9,
827
827
  config_enum: str = 'parallel',
828
828
  threshold: int = 0,
829
- use_jax: bool = False):
829
+ use_jax: bool = True):
830
830
  """
831
831
  Train the SURE model.
832
832
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.2.1
3
+ Version: 2.2.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.2.1',
8
+ version='2.2.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