change default emb to cls
Browse files
geneformer/emb_extractor.py
CHANGED
|
@@ -411,7 +411,7 @@ class EmbExtractor:
|
|
| 411 |
self,
|
| 412 |
model_type="Pretrained",
|
| 413 |
num_classes=0,
|
| 414 |
-
emb_mode="
|
| 415 |
cell_emb_style="mean_pool",
|
| 416 |
gene_emb_style="mean_pool",
|
| 417 |
filter_data=None,
|
|
|
|
| 411 |
self,
|
| 412 |
model_type="Pretrained",
|
| 413 |
num_classes=0,
|
| 414 |
+
emb_mode="cls",
|
| 415 |
cell_emb_style="mean_pool",
|
| 416 |
gene_emb_style="mean_pool",
|
| 417 |
filter_data=None,
|