SURE-tools 2.1.7__py3-none-any.whl → 2.1.8__py3-none-any.whl

SURE/perturb/__init__.py

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

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.dist-info → sure_tools-2.1.8.dist-info}/METADATA

@@ -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.dist-info → sure_tools-2.1.8.dist-info}/RECORD

@@ -13,15 +13,15 @@ SURE/flow/__init__.py,sha256=eIcHeeISEOXEB0zRW-uiRlDz5IEUbA7rZZ1htUxhLVU,167
 SURE/flow/flow_stats.py,sha256=wpIxmOpYPU7YcgxqXKS8EDmnZ3oGh-YdBjiBiEmHcoU,2360
 SURE/flow/plot_quiver.py,sha256=spcPC0BDpvC-FBk9XIu6rcJI64bS-U-IVmmKLYHHXgs,1865
 SURE/flow/quiver.py,sha256=_euFqSaRrDoZ_oOabOx20LOoUTJ__XPhLW-vzLNQfAo,1859
-SURE/perturb/__init__.py,sha256=vSQMtxvAuLkaYweg6MM87W9grJu9X9HGvfW3WRBVbAI,108
-SURE/perturb/perturb.py,sha256=ZHIeOcm64SYXnbTH5hLxVgRX4DiPZgzJQu7UMkeZ6zE,2460
+SURE/perturb/__init__.py,sha256=ouxShhbxZM4r5Gf7GmKiutrsmtyq7QL8rHjhgF0BU08,32
+SURE/perturb/perturb.py,sha256=LhrQSZEka7GFc9x4YfMhzHEYE-oHWwDg5tlhIRXVtGg,3237
 SURE/utils/__init__.py,sha256=Htqv4KqVKcRiaaTBsR-6yZ4LSlbhbzutjNKXGD9-uds,660
 SURE/utils/custom_mlp.py,sha256=07TYX1HgxfEjb_3i5MpiZfNhOhx3dKntuwGkrpteWiM,7036
 SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
 SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
-sure_tools-2.1.7.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
-sure_tools-2.1.7.dist-info/METADATA,sha256=5V5tWo33sIhQbAL0LbZLWIlM3J-TRIdS3zRpUvFEIMk,2650
-sure_tools-2.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sure_tools-2.1.7.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
-sure_tools-2.1.7.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
-sure_tools-2.1.7.dist-info/RECORD,,
+sure_tools-2.1.8.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
+sure_tools-2.1.8.dist-info/METADATA,sha256=DJqLBtteAneDszgl4HiVf-Fucqzn9thAXoXzPq6b6_Y,2650
+sure_tools-2.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sure_tools-2.1.8.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
+sure_tools-2.1.8.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
+sure_tools-2.1.8.dist-info/RECORD,,