ctheodoris
/

Geneformer

Model card Files Files and versions

ctheodoris commited on Oct 5, 2024

Commit

9ee140a

·

verified ·

1 Parent(s): 21d36cd

change default emb to cls

Files changed (1) hide show

geneformer/emb_extractor.py +1 -1

geneformer/emb_extractor.py CHANGED Viewed

@@ -411,7 +411,7 @@ class EmbExtractor:
         self,
         model_type="Pretrained",
         num_classes=0,
-        emb_mode="cell",
         cell_emb_style="mean_pool",
         gene_emb_style="mean_pool",
         filter_data=None,

         self,
         model_type="Pretrained",
         num_classes=0,
+        emb_mode="cls",
         cell_emb_style="mean_pool",
         gene_emb_style="mean_pool",
         filter_data=None,