Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/PharMolix/SToFM
/ functions
Functions
139 in github.com/PharMolix/SToFM
⨍
Functions
139
◇
Types & classes
32
↓ 12 callers
Method
normal_
(self, data: torch.Tensor)
model/se2transformer.py:352
↓ 6 callers
Method
__init__
(self, config: SToFMConfig)
model/se2transformer.py:416
↓ 5 callers
Function
flatten_list
(megalist)
geneformer_001/geneformer/in_silico_perturber.py:91
↓ 4 callers
Function
gen_attention_mask
(minibatch_encoding, max_len = None)
geneformer_001/geneformer/in_silico_perturber.py:568
↓ 4 callers
Function
mean_nonpadding_embs
(embs, original_lens)
geneformer_001/geneformer/in_silico_perturber.py:580
↓ 4 callers
Function
quant_cos_sims
(model, perturb_type, perturbation_batch, forward_b
geneformer_001/geneformer/in_silico_perturber.py:310
↓ 3 callers
Function
downsample_and_sort
(data_shuffled, max_ncells)
geneformer_001/geneformer/in_silico_perturber.py:98
↓ 3 callers
Function
encode_cell
(model, model_input_path, emb_path, save=True, batch_size=1, add_cls=True)
model/extraction.py:53
↓ 3 callers
Function
get_fdr
(pvalues)
geneformer_001/geneformer/in_silico_perturber_stats.py:99
↓ 3 callers
Function
get_impact_component
(test_value, gaussian_mixture_model)
geneformer_001/geneformer/in_silico_perturber_stats.py:102
↓ 3 callers
Function
get_model_input_size
(model)
geneformer_001/geneformer/in_silico_perturber.py:88
↓ 3 callers
Function
get_possible_states
(cell_states_to_model)
geneformer_001/geneformer/in_silico_perturber.py:108
↓ 3 callers
Function
make_perturbation_batch
(example_cell, perturb_type, tokens_to_perturb,
geneformer_001/geneformer/in_silico_perturber.py:171
↓ 3 callers
Method
pad
Pad a single encoded input or a batch of encoded inputs up to predefined length or to the max sequence length in the batch.
geneformer_001/geneformer/pretrainer.py:273
↓ 3 callers
Function
read_dictionaries
(input_data_directory, cell_or_gene_emb, anchor_token)
geneformer_001/geneformer/in_silico_perturber_stats.py:45
↓ 2 callers
Method
_convert_token_to_id_with_added_voc
(self, token)
geneformer_001/geneformer/collator_for_classification.py:512
↓ 2 callers
Method
_convert_token_to_id_with_added_voc
(self, token)
geneformer_001/geneformer/pretrainer.py:587
↓ 2 callers
Method
_pad
Pad encoded inputs (on left/right and up to predefined length or max length in the batch) Args: encoded_inputs: Dictiona
geneformer_001/geneformer/collator_for_classification.py:386
↓ 2 callers
Method
_pad
Pad encoded inputs (on left/right and up to predefined length or max length in the batch) Args: encoded_inputs: Dictiona
geneformer_001/geneformer/pretrainer.py:439
↓ 2 callers
Method
_prepare_batch
(self, features)
geneformer_001/geneformer/collator_for_classification.py:565
↓ 2 callers
Function
cos_sim_shift
(original_emb, minibatch_emb, end_emb, perturb_group,
geneformer_001/geneformer/in_silico_perturber.py:474
↓ 2 callers
Method
create_dataset
(self, tokenized_cells, cell_metadata)
geneformer_001/geneformer/tokenizer.py:210
↓ 2 callers
Function
delete_indices
(example)
geneformer_001/geneformer/in_silico_perturber.py:131
↓ 2 callers
Function
get_gene_list
(dict_list,mode)
geneformer_001/geneformer/in_silico_perturber_stats.py:73
↓ 2 callers
Function
get_hypernodes
(nodes, features, alpha=0.2, device=-1, leiden_res=1.0)
model/extraction.py:73
↓ 2 callers
Function
get_input
(nodes, features, indices, token_types, device=-1, hyper_type=2)
model/extraction.py:165
↓ 2 callers
Function
invert_dict
(dictionary)
geneformer_001/geneformer/in_silico_perturber_stats.py:41
↓ 2 callers
Function
load_and_filter
(filter_data, nproc, input_data_file)
geneformer_001/geneformer/in_silico_perturber.py:46
↓ 2 callers
Function
load_model
(model_type, num_classes, model_directory)
geneformer_001/geneformer/in_silico_perturber.py:61
↓ 2 callers
Function
n_detections
(token, dict_list, mode, anchor_token)
geneformer_001/geneformer/in_silico_perturber_stats.py:90
↓ 2 callers
Function
pad_or_truncate_encoding
(encoding, pad_token_id, max_len)
geneformer_001/geneformer/in_silico_perturber.py:534
↓ 2 callers
Function
pad_tensor
(tensor, pad_token_id, max_len)
geneformer_001/geneformer/in_silico_perturber.py:517
↓ 2 callers
Function
pad_tensor_list
(tensor_list, dynamic_or_constant, pad_token_id, model_input_size)
geneformer_001/geneformer/in_silico_perturber.py:549
↓ 2 callers
Function
quant_layers
(model)
geneformer_001/geneformer/in_silico_perturber.py:81
↓ 2 callers
Function
split_graph
(nodes, features, num=1000)
model/extraction.py:119
↓ 2 callers
Function
tokenize_cell
Convert normalized gene expression vector to tokenized rank value encoding.
geneformer_001/geneformer/tokenizer.py:35
↓ 1 callers
Method
__init__
(self, *args, **kwargs)
geneformer_001/geneformer/pretrainer.py:598
↓ 1 callers
Method
__init__
( self, num_hidden_layers: int = 12, embedding_dim: int = 768, ffn_embedding_d
model/utils.py:58
↓ 1 callers
Method
_get_padding_truncation_strategies
Find the correct padding/truncation strategy with backward compatibility for old arguments (truncation_strategy and pad_to_max_length
geneformer_001/geneformer/collator_for_classification.py:94
↓ 1 callers
Method
_get_padding_truncation_strategies
Find the correct padding/truncation strategy with backward compatibility for old arguments (truncation_strategy and pad_to_max_length
geneformer_001/geneformer/pretrainer.py:127
↓ 1 callers
Method
apply_sparse_mask
(self, attn_weights: torch.Tensor, tgt_len: int, src_len: int, bsz: int)
model/se2transformer.py:208
↓ 1 callers
Function
forward_pass_single_cell
(model, example_cell, layer_to_quant)
geneformer_001/geneformer/in_silico_perturber.py:115
↓ 1 callers
Function
gen_heatmap_class_colors
(labels, df)
geneformer_001/geneformer/emb_extractor.py:135
↓ 1 callers
Function
gen_heatmap_class_dict
(classes, label_colors_series)
geneformer_001/geneformer/emb_extractor.py:141
↓ 1 callers
Function
get_cell_state_avg_embs
(model, filtered_input_data, cell_states_to_model,
geneformer_001/geneformer/in_silico_perturber.py:251
↓ 1 callers
Function
get_embs
(model, filtered_input_data, emb_mode, layer_to_quant, pad
geneformer_001/geneformer/emb_extractor.py:58
↓ 1 callers
Function
get_length_grouped_indices
Return a list of indices so that each slice of :obj:`batch_size` consecutive indices correspond to elements of similar lengths. To do this, t
geneformer_001/geneformer/pretrainer.py:778
↓ 1 callers
Method
in_silico_perturb
(self, model, filtered_input_data,
geneformer_001/geneformer/in_silico_perturber.py:983
↓ 1 callers
Function
isp_aggregate_grouped_perturb
(cos_sims_df, dict_list)
geneformer_001/geneformer/in_silico_perturber_stats.py:118
↓ 1 callers
Function
isp_stats_mixture_model
(cos_sims_df, dict_list, combos, anchor_token)
geneformer_001/geneformer/in_silico_perturber_stats.py:296
↓ 1 callers
Function
isp_stats_to_goal_state
(cos_sims_df, dict_list, cell_states_to_model, genes_perturbed)
geneformer_001/geneformer/in_silico_perturber_stats.py:130
↓ 1 callers
Function
isp_stats_vs_null
(cos_sims_df, dict_list, null_dict_list)
geneformer_001/geneformer/in_silico_perturber_stats.py:251
↓ 1 callers
Function
label_embs
(embs, downsampled_data, emb_labels)
geneformer_001/geneformer/emb_extractor.py:107
↓ 1 callers
Function
load_data
(data_path, spatial_path=None, emb_path=None, new_emb=False, model=None, model_input_path=None,
model/extraction.py:215
↓ 1 callers
Function
make_colorbar
(embs_df, label)
geneformer_001/geneformer/emb_extractor.py:146
↓ 1 callers
Function
make_comparison_batch
(original_emb_batch, indices_to_perturb, perturb_group)
geneformer_001/geneformer/in_silico_perturber.py:218
↓ 1 callers
Method
mask_attn_bias
(self, attn_bias, node_coops, token_types)
model/extraction.py:388
↓ 1 callers
Method
mask_nodes
(self, token_embeddings, token_types)
model/extraction.py:371
↓ 1 callers
Function
overexpress_tokens
(example)
geneformer_001/geneformer/in_silico_perturber.py:149
↓ 1 callers
Method
pad
Pad a single encoded input or a batch of encoded inputs up to predefined length or to the max sequence length in the batch.
geneformer_001/geneformer/collator_for_classification.py:222
↓ 1 callers
Function
pad_2d_tensor
(tensor, pad_token_id, max_len, dim)
geneformer_001/geneformer/in_silico_perturber.py:524
↓ 1 callers
Function
pad_list
(input_ids, pad_token_id, max_len)
geneformer_001/geneformer/in_silico_perturber.py:511
↓ 1 callers
Function
plot_heatmap
(embs_df, emb_dims, label, output_file, kwargs_dict)
geneformer_001/geneformer/emb_extractor.py:164
↓ 1 callers
Function
plot_umap
(embs_df, emb_dims, label, output_file, kwargs_dict)
geneformer_001/geneformer/emb_extractor.py:115
↓ 1 callers
Function
preprocess
(nodes, features, num=1000, device=-1, leiden_res=1.0, alpha=0.2)
model/extraction.py:160
↓ 1 callers
Function
remove_indices_from_emb
(emb, indices_to_remove, gene_dim)
geneformer_001/geneformer/in_silico_perturber.py:156
↓ 1 callers
Function
remove_indices_from_emb_batch
(emb_batch, list_of_indices_to_remove, gene_dim)
geneformer_001/geneformer/in_silico_perturber.py:164
↓ 1 callers
Method
reset_parameters
(self)
model/se2transformer.py:87
↓ 1 callers
Method
token_to_gene_name
(self, item)
geneformer_001/geneformer/in_silico_perturber_stats.py:712
↓ 1 callers
Function
token_tuple_to_ensembl_ids
(token_tuple, gene_token_id_dict)
geneformer_001/geneformer/in_silico_perturber_stats.py:87
↓ 1 callers
Method
tokenize_anndata
(self, data)
preprocessing/preprocess.py:9
↓ 1 callers
Method
tokenize_file
(self, loom_file_path)
geneformer_001/geneformer/tokenizer.py:141
↓ 1 callers
Method
tokenize_files
(self, loom_data_directory)
geneformer_001/geneformer/tokenizer.py:114
↓ 1 callers
Method
validate_options
(self)
geneformer_001/geneformer/in_silico_perturber_stats.py:491
↓ 1 callers
Method
validate_options
(self)
geneformer_001/geneformer/in_silico_perturber.py:774
↓ 1 callers
Method
validate_options
(self)
geneformer_001/geneformer/emb_extractor.py:291
Method
__call__
(self, features)
geneformer_001/geneformer/collator_for_classification.py:578
Method
__call__
(self, features)
model/extraction.py:432
Method
__init__
(self, config, pretrained_proj, proj_dim)
get_embeddings.py:52
Method
__init__
Initialize in silico perturber stats generator. Parameters ---------- mode : {"goal_state_shift","vs_null","mixture_
geneformer_001/geneformer/in_silico_perturber_stats.py:420
Method
__init__
Initialize in silico perturber. Parameters ---------- perturb_type : {"delete","overexpress","inhibit","activate"}
geneformer_001/geneformer/in_silico_perturber.py:613
Method
__init__
Initialize tokenizer. Parameters ---------- custom_attr_name_dict : None, dict Dictionary of custom attr
geneformer_001/geneformer/tokenizer.py:50
Method
__init__
(self, *args, **kwargs)
geneformer_001/geneformer/collator_for_classification.py:556
Method
__init__
Initialize embedding extractor. Parameters ---------- model_type : {"Pretrained","GeneClassifier","CellClassifier"}
geneformer_001/geneformer/emb_extractor.py:212
Method
__init__
(self, *args, **kwargs)
geneformer_001/geneformer/pretrainer.py:108
Method
__init__
( self, dataset: Dataset, batch_size: int, num_replicas: Optional[int] = None,
geneformer_001/geneformer/pretrainer.py:698
Method
__init__
(self, *args, **kwargs)
preprocessing/preprocess.py:6
Method
__init__
(self, norm_type=0, cls_type=1, hyper_type=2, pad_type=3, pad_indices=-1, mask=False, mask_ra
model/extraction.py:349
Method
__init__
(self, add_cls=True, *args, **kwargs)
model/utils.py:28
Method
__init__
(self, p: float, modules: Optional[Iterable[nn.Module]] = None)
model/se2transformer.py:18
Method
__init__
(self, config: SToFMConfig)
model/se2transformer.py:29
Method
__init__
(self, config: SToFMConfig)
model/se2transformer.py:61
Method
__init__
(self, config: SToFMConfig)
model/se2transformer.py:213
Method
__init__
(self, config: SToFMConfig)
model/se2transformer.py:283
Method
__init__
(self, config: SToFMConfig)
model/se2transformer.py:460
Method
__iter__
(self)
geneformer_001/geneformer/pretrainer.py:756
Method
__iter__
(self)
model/se2transformer.py:22
Method
__len__
(self)
geneformer_001/geneformer/collator_for_classification.py:518
Method
__len__
(self)
geneformer_001/geneformer/pretrainer.py:593
Method
__len__
(self)
model/utils.py:24
next →
1–100 of 139, ranked by callers