SURE-tools 2.1.92__tar.gz → 2.2.1__tar.gz

{sure_tools-2.1.92 → sure_tools-2.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SURE-tools
-Version: 2.1.92
+Version: 2.2.1
 Summary: Succinct Representation of Single Cells
 Home-page: https://github.com/ZengFLab/SURE
 Author: Feng Zeng

sure_tools-2.1.92/SURE/PerturbFlow.py → sure_tools-2.2.1/SURE/DensityFlow.py

@@ -54,7 +54,7 @@ def set_random_seed(seed):
     # Set seed for Pyro
     pyro.set_rng_seed(seed)
 
-class PerturbFlow(nn.Module):
+class DensityFlow(nn.Module):
     def __init__(self,
                  input_size: int,
                  codebook_size: int = 200,
@@ -308,7 +308,7 @@ class PerturbFlow(nn.Module):
         return xs
 
     def model1(self, xs):
-        pyro.module('PerturbFlow', self)
+        pyro.module('DensityFlow', self)
 
         eps = torch.finfo(xs.dtype).eps
         batch_size = xs.size(0)
@@ -387,7 +387,7 @@ class PerturbFlow(nn.Module):
             ns = pyro.sample('n', dist.OneHotCategorical(logits=alpha))
 
     def model2(self, xs, us=None):
-        pyro.module('PerturbFlow', self)
+        pyro.module('DensityFlow', self)
 
         eps = torch.finfo(xs.dtype).eps
         batch_size = xs.size(0)
@@ -471,7 +471,7 @@ class PerturbFlow(nn.Module):
             ns = pyro.sample('n', dist.OneHotCategorical(logits=alpha))
 
     def model3(self, xs, ys, embeds=None):
-        pyro.module('PerturbFlow', self)
+        pyro.module('DensityFlow', self)
 
         eps = torch.finfo(xs.dtype).eps
         batch_size = xs.size(0)
@@ -567,7 +567,7 @@ class PerturbFlow(nn.Module):
                 zns = embeds
 
     def model4(self, xs, us, ys, embeds=None):
-        pyro.module('PerturbFlow', self)
+        pyro.module('DensityFlow', self)
 
         eps = torch.finfo(xs.dtype).eps
         batch_size = xs.size(0)
@@ -970,7 +970,7 @@ class PerturbFlow(nn.Module):
             us = None, 
             ys = None,
             zs = None,
-            num_epochs: int = 200, 
+            num_epochs: int = 500, 
             learning_rate: float = 0.0001, 
             batch_size: int = 256, 
             algo: Literal['adam','rmsprop','adamw'] = 'adam', 
@@ -981,7 +981,7 @@ class PerturbFlow(nn.Module):
             threshold: int = 0,
             use_jax: bool = True):
         """
-        Train the PerturbFlow model.
+        Train the DensityFlow model.
 
         Parameters
         ----------
@@ -1007,7 +1007,7 @@ class PerturbFlow(nn.Module):
             Parameter for optimization.
         use_jax
             If toggled on, Jax will be used for speeding up. CAUTION: This will raise errors because of unknown reasons when it is called in
-            the Python script or Jupyter notebook. It is OK if it is used when runing PerturbFlow in the shell command.
+            the Python script or Jupyter notebook. It is OK if it is used when runing DensityFlow in the shell command.
         """
         xs = self.preprocess(xs, threshold=threshold)
         xs = convert_to_tensor(xs, dtype=self.dtype, device=self.get_device())
@@ -1125,12 +1125,12 @@ class PerturbFlow(nn.Module):
 
         
 EXAMPLE_RUN = (
-    "example run: PerturbFlow --help"
+    "example run: DensityFlow --help"
 )
 
 def parse_args():
     parser = argparse.ArgumentParser(
-        description="PerturbFlow\n{}".format(EXAMPLE_RUN))
+        description="DensityFlow\n{}".format(EXAMPLE_RUN))
 
     parser.add_argument(
         "--cuda", action="store_true", help="use GPU(s) to speed up training"
@@ -1317,7 +1317,7 @@ def main():
     cell_factor_size = 0 if us is None else us.shape[1]
 
     ###########################################
-    perturbflow = PerturbFlow(
+    DensityFlow = DensityFlow(
         input_size=input_size,
         cell_factor_size=cell_factor_size,
         inverse_dispersion=args.inverse_dispersion,
@@ -1336,7 +1336,7 @@ def main():
         dtype=dtype,
     )
 
-    perturbflow.fit(xs, us=us, 
+    DensityFlow.fit(xs, us=us, 
              num_epochs=args.num_epochs,
              learning_rate=args.learning_rate,
              batch_size=args.batch_size,
@@ -1348,9 +1348,9 @@ def main():
 
     if args.save_model is not None:
         if args.save_model.endswith('gz'):
-            PerturbFlow.save_model(perturbflow, args.save_model, compression=True)
+            DensityFlow.save_model(DensityFlow, args.save_model, compression=True)
         else:
-            PerturbFlow.save_model(perturbflow, args.save_model)
+            DensityFlow.save_model(DensityFlow, args.save_model)
     
 
 

{sure_tools-2.1.92 → sure_tools-2.2.1}/SURE/__init__.py

@@ -1,12 +1,12 @@
 from .SURE import SURE
-from .PerturbFlow import PerturbFlow
+from .DensityFlow import DensityFlow
 
 from . import utils 
 from . import codebook
 from . import SURE
-from . import PerturbFlow
+from . import DensityFlow
 from . import atac
 from . import flow 
 from . import perturb
 
-__all__ = ['SURE', 'PerturbFlow', 'flow', 'perturb', 'atac', 'utils', 'codebook']
+__all__ = ['SURE', 'DensityFlow', 'flow', 'perturb', 'atac', 'utils', 'codebook']

{sure_tools-2.1.92 → sure_tools-2.2.1}/SURE/flow/flow_stats.py

@@ -41,6 +41,18 @@ class VectorFieldEval:
         divergence[np.isnan(divergence)] = 0
         
         return divergence
+    
+    def movement_stats(self,vectors):
+        return calculate_movement_stats(vectors)
+    
+    def direction_stats(self, vectors):
+        return calculate_direction_stats(vectors)
+    
+    def movement_energy(self, vectors, masses=None):
+        return calculate_movement_energy(vectors, masses)
+    
+    def movement_divergence(self, positions, vectors):
+        return calculate_movement_divergence(positions, vectors)
 
         
 def calculate_movement_stats(vectors):

{sure_tools-2.1.92 → sure_tools-2.2.1}/SURE_tools.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SURE-tools
-Version: 2.1.92
+Version: 2.2.1
 Summary: Succinct Representation of Single Cells
 Home-page: https://github.com/ZengFLab/SURE
 Author: Feng Zeng

{sure_tools-2.1.92 → sure_tools-2.2.1}/SURE_tools.egg-info/SOURCES.txt

@@ -1,7 +1,7 @@
 LICENSE
 README.md
 setup.py
-SURE/PerturbFlow.py
+SURE/DensityFlow.py
 SURE/SURE.py
 SURE/__init__.py
 SURE/assembly/__init__.py

{sure_tools-2.1.92 → sure_tools-2.2.1}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setup(
     name='SURE-tools',
-    version='2.1.92',
+    version='2.2.1',
     description='Succinct Representation of Single Cells',
     long_description=long_description,
     long_description_content_type="text/markdown",