SURE-tools 2.1.7__tar.gz → 2.1.8__tar.gz

{sure_tools-2.1.7 → sure_tools-2.1.8}/PKG-INFO

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

sure_tools-2.1.8/SURE/perturb/__init__.py

@@ -0,0 +1 @@
+from .perturb import LabelMatrix

{sure_tools-2.1.7 → sure_tools-2.1.8}/SURE/perturb/perturb.py

@@ -2,7 +2,24 @@ import re
 import numpy as np
 from itertools import chain
 from joblib import Parallel, delayed
+from typing import Literal
 
+class LabelMatrix:
+    def __init__(self):
+        self.labels_ = None
+    
+    def fit_transform(self, labels, sep_pattern=r'[;_\-\s]', speedup: Literal['none','vectorize','parallel']='none'):
+        if speedup=='none':
+            mat, self.labels_ = label_to_matrix(labels=labels, sep_pattern=sep_pattern)
+        elif speedup=='vectorize':
+            mat, self.labels_ = vectorized_label_to_matrix(labels=labels, sep_pattern=sep_pattern)
+        elif speedup=='parallel':
+            mat, self.labels_ = parallel_label_to_matrix(labels=labels, sep_pattern=sep_pattern)
+        return mat
+            
+    def inverse_transform(self, matrix):
+        matrix_to_labels(matrix=matrix, unique_labels=self.labels_)
+        
 def label_to_matrix(labels, sep_pattern=r'[;_\-\s]'):
     """
     将混合分隔符的多标签数据转换为 0-1 矩阵
@@ -48,8 +65,8 @@ def vectorized_label_to_matrix(labels, sep_pattern=r'[;_\-\s]'):
 
 
 
-def parallel_label_to_matrix(labels, n_jobs=4):
-    labels_unified = [re.sub(r'[;_\-\s]', ';', label) for label in labels]
+def parallel_label_to_matrix(labels, sep_pattern=r'[;_\-\s]', n_jobs=4):
+    labels_unified = [re.sub(sep_pattern, ';', label) for label in labels]
     unique_labels = sorted(set(chain(*[label.split(';') for label in labels_unified])))
     
     def process_row(row_labels, unique_labels):

{sure_tools-2.1.7 → sure_tools-2.1.8}/SURE_tools.egg-info/PKG-INFO

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

{sure_tools-2.1.7 → sure_tools-2.1.8}/setup.py

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

sure_tools-2.1.7/SURE/perturb/__init__.py

@@ -1 +0,0 @@
-from .perturb import label_to_matrix, matrix_to_labels, vectorized_label_to_matrix, parallel_label_to_matrix