Christina Theodoris
commited on
Commit
·
f45a6c7
1
Parent(s):
be98543
default id_class_dict to None for gene classifiers
Browse files
geneformer/classifier_utils.py
CHANGED
|
@@ -94,7 +94,7 @@ def remove_rare(data, rare_threshold, label, nproc):
|
|
| 94 |
return data
|
| 95 |
|
| 96 |
|
| 97 |
-
def label_classes(classifier, data, gene_class_dict, nproc, id_class_dict):
|
| 98 |
if classifier == "cell":
|
| 99 |
label_set = set(data["label"])
|
| 100 |
elif classifier == "gene":
|
|
|
|
| 94 |
return data
|
| 95 |
|
| 96 |
|
| 97 |
+
def label_classes(classifier, data, gene_class_dict, nproc, id_class_dict=None):
|
| 98 |
if classifier == "cell":
|
| 99 |
label_set = set(data["label"])
|
| 100 |
elif classifier == "gene":
|