ctheodoris commited on
Commit
0c99403
·
verified ·
1 Parent(s): 3a68669

update isp default mode to cls

Browse files
Files changed (1) hide show
  1. geneformer/in_silico_perturber.py +1 -1
geneformer/in_silico_perturber.py CHANGED
@@ -84,7 +84,7 @@ class InSilicoPerturber:
84
  anchor_gene=None,
85
  model_type="Pretrained",
86
  num_classes=0,
87
- emb_mode="cell",
88
  cell_emb_style="mean_pool",
89
  filter_data=None,
90
  cell_states_to_model=None,
 
84
  anchor_gene=None,
85
  model_type="Pretrained",
86
  num_classes=0,
87
+ emb_mode="cls",
88
  cell_emb_style="mean_pool",
89
  filter_data=None,
90
  cell_states_to_model=None,