MCPcopy Create free account

hub / github.com/PharMolix/SToFM / functions

Functions139 in github.com/PharMolix/SToFM

↓ 12 callersMethodnormal_
(self, data: torch.Tensor)
model/se2transformer.py:352
↓ 6 callersMethod__init__
(self, config: SToFMConfig)
model/se2transformer.py:416
↓ 5 callersFunctionflatten_list
(megalist)
geneformer_001/geneformer/in_silico_perturber.py:91
↓ 4 callersFunctiongen_attention_mask
(minibatch_encoding, max_len = None)
geneformer_001/geneformer/in_silico_perturber.py:568
↓ 4 callersFunctionmean_nonpadding_embs
(embs, original_lens)
geneformer_001/geneformer/in_silico_perturber.py:580
↓ 4 callersFunctionquant_cos_sims
(model, perturb_type, perturbation_batch, forward_b
geneformer_001/geneformer/in_silico_perturber.py:310
↓ 3 callersFunctiondownsample_and_sort
(data_shuffled, max_ncells)
geneformer_001/geneformer/in_silico_perturber.py:98
↓ 3 callersFunctionencode_cell
(model, model_input_path, emb_path, save=True, batch_size=1, add_cls=True)
model/extraction.py:53
↓ 3 callersFunctionget_fdr
(pvalues)
geneformer_001/geneformer/in_silico_perturber_stats.py:99
↓ 3 callersFunctionget_impact_component
(test_value, gaussian_mixture_model)
geneformer_001/geneformer/in_silico_perturber_stats.py:102
↓ 3 callersFunctionget_model_input_size
(model)
geneformer_001/geneformer/in_silico_perturber.py:88
↓ 3 callersFunctionget_possible_states
(cell_states_to_model)
geneformer_001/geneformer/in_silico_perturber.py:108
↓ 3 callersFunctionmake_perturbation_batch
(example_cell, perturb_type, tokens_to_perturb,
geneformer_001/geneformer/in_silico_perturber.py:171
↓ 3 callersMethodpad
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 callersFunctionread_dictionaries
(input_data_directory, cell_or_gene_emb, anchor_token)
geneformer_001/geneformer/in_silico_perturber_stats.py:45
↓ 2 callersMethod_convert_token_to_id_with_added_voc
(self, token)
geneformer_001/geneformer/collator_for_classification.py:512
↓ 2 callersMethod_convert_token_to_id_with_added_voc
(self, token)
geneformer_001/geneformer/pretrainer.py:587
↓ 2 callersMethod_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 callersMethod_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 callersMethod_prepare_batch
(self, features)
geneformer_001/geneformer/collator_for_classification.py:565
↓ 2 callersFunctioncos_sim_shift
(original_emb, minibatch_emb, end_emb, perturb_group,
geneformer_001/geneformer/in_silico_perturber.py:474
↓ 2 callersMethodcreate_dataset
(self, tokenized_cells, cell_metadata)
geneformer_001/geneformer/tokenizer.py:210
↓ 2 callersFunctiondelete_indices
(example)
geneformer_001/geneformer/in_silico_perturber.py:131
↓ 2 callersFunctionget_gene_list
(dict_list,mode)
geneformer_001/geneformer/in_silico_perturber_stats.py:73
↓ 2 callersFunctionget_hypernodes
(nodes, features, alpha=0.2, device=-1, leiden_res=1.0)
model/extraction.py:73
↓ 2 callersFunctionget_input
(nodes, features, indices, token_types, device=-1, hyper_type=2)
model/extraction.py:165
↓ 2 callersFunctioninvert_dict
(dictionary)
geneformer_001/geneformer/in_silico_perturber_stats.py:41
↓ 2 callersFunctionload_and_filter
(filter_data, nproc, input_data_file)
geneformer_001/geneformer/in_silico_perturber.py:46
↓ 2 callersFunctionload_model
(model_type, num_classes, model_directory)
geneformer_001/geneformer/in_silico_perturber.py:61
↓ 2 callersFunctionn_detections
(token, dict_list, mode, anchor_token)
geneformer_001/geneformer/in_silico_perturber_stats.py:90
↓ 2 callersFunctionpad_or_truncate_encoding
(encoding, pad_token_id, max_len)
geneformer_001/geneformer/in_silico_perturber.py:534
↓ 2 callersFunctionpad_tensor
(tensor, pad_token_id, max_len)
geneformer_001/geneformer/in_silico_perturber.py:517
↓ 2 callersFunctionpad_tensor_list
(tensor_list, dynamic_or_constant, pad_token_id, model_input_size)
geneformer_001/geneformer/in_silico_perturber.py:549
↓ 2 callersFunctionquant_layers
(model)
geneformer_001/geneformer/in_silico_perturber.py:81
↓ 2 callersFunctionsplit_graph
(nodes, features, num=1000)
model/extraction.py:119
↓ 2 callersFunctiontokenize_cell
Convert normalized gene expression vector to tokenized rank value encoding.
geneformer_001/geneformer/tokenizer.py:35
↓ 1 callersMethod__init__
(self, *args, **kwargs)
geneformer_001/geneformer/pretrainer.py:598
↓ 1 callersMethod__init__
( self, num_hidden_layers: int = 12, embedding_dim: int = 768, ffn_embedding_d
model/utils.py:58
↓ 1 callersMethod_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 callersMethod_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 callersMethodapply_sparse_mask
(self, attn_weights: torch.Tensor, tgt_len: int, src_len: int, bsz: int)
model/se2transformer.py:208
↓ 1 callersFunctionforward_pass_single_cell
(model, example_cell, layer_to_quant)
geneformer_001/geneformer/in_silico_perturber.py:115
↓ 1 callersFunctiongen_heatmap_class_colors
(labels, df)
geneformer_001/geneformer/emb_extractor.py:135
↓ 1 callersFunctiongen_heatmap_class_dict
(classes, label_colors_series)
geneformer_001/geneformer/emb_extractor.py:141
↓ 1 callersFunctionget_cell_state_avg_embs
(model, filtered_input_data, cell_states_to_model,
geneformer_001/geneformer/in_silico_perturber.py:251
↓ 1 callersFunctionget_embs
(model, filtered_input_data, emb_mode, layer_to_quant, pad
geneformer_001/geneformer/emb_extractor.py:58
↓ 1 callersFunctionget_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 callersMethodin_silico_perturb
(self, model, filtered_input_data,
geneformer_001/geneformer/in_silico_perturber.py:983
↓ 1 callersFunctionisp_aggregate_grouped_perturb
(cos_sims_df, dict_list)
geneformer_001/geneformer/in_silico_perturber_stats.py:118
↓ 1 callersFunctionisp_stats_mixture_model
(cos_sims_df, dict_list, combos, anchor_token)
geneformer_001/geneformer/in_silico_perturber_stats.py:296
↓ 1 callersFunctionisp_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 callersFunctionisp_stats_vs_null
(cos_sims_df, dict_list, null_dict_list)
geneformer_001/geneformer/in_silico_perturber_stats.py:251
↓ 1 callersFunctionlabel_embs
(embs, downsampled_data, emb_labels)
geneformer_001/geneformer/emb_extractor.py:107
↓ 1 callersFunctionload_data
(data_path, spatial_path=None, emb_path=None, new_emb=False, model=None, model_input_path=None,
model/extraction.py:215
↓ 1 callersFunctionmake_colorbar
(embs_df, label)
geneformer_001/geneformer/emb_extractor.py:146
↓ 1 callersFunctionmake_comparison_batch
(original_emb_batch, indices_to_perturb, perturb_group)
geneformer_001/geneformer/in_silico_perturber.py:218
↓ 1 callersMethodmask_attn_bias
(self, attn_bias, node_coops, token_types)
model/extraction.py:388
↓ 1 callersMethodmask_nodes
(self, token_embeddings, token_types)
model/extraction.py:371
↓ 1 callersFunctionoverexpress_tokens
(example)
geneformer_001/geneformer/in_silico_perturber.py:149
↓ 1 callersMethodpad
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 callersFunctionpad_2d_tensor
(tensor, pad_token_id, max_len, dim)
geneformer_001/geneformer/in_silico_perturber.py:524
↓ 1 callersFunctionpad_list
(input_ids, pad_token_id, max_len)
geneformer_001/geneformer/in_silico_perturber.py:511
↓ 1 callersFunctionplot_heatmap
(embs_df, emb_dims, label, output_file, kwargs_dict)
geneformer_001/geneformer/emb_extractor.py:164
↓ 1 callersFunctionplot_umap
(embs_df, emb_dims, label, output_file, kwargs_dict)
geneformer_001/geneformer/emb_extractor.py:115
↓ 1 callersFunctionpreprocess
(nodes, features, num=1000, device=-1, leiden_res=1.0, alpha=0.2)
model/extraction.py:160
↓ 1 callersFunctionremove_indices_from_emb
(emb, indices_to_remove, gene_dim)
geneformer_001/geneformer/in_silico_perturber.py:156
↓ 1 callersFunctionremove_indices_from_emb_batch
(emb_batch, list_of_indices_to_remove, gene_dim)
geneformer_001/geneformer/in_silico_perturber.py:164
↓ 1 callersMethodreset_parameters
(self)
model/se2transformer.py:87
↓ 1 callersMethodtoken_to_gene_name
(self, item)
geneformer_001/geneformer/in_silico_perturber_stats.py:712
↓ 1 callersFunctiontoken_tuple_to_ensembl_ids
(token_tuple, gene_token_id_dict)
geneformer_001/geneformer/in_silico_perturber_stats.py:87
↓ 1 callersMethodtokenize_anndata
(self, data)
preprocessing/preprocess.py:9
↓ 1 callersMethodtokenize_file
(self, loom_file_path)
geneformer_001/geneformer/tokenizer.py:141
↓ 1 callersMethodtokenize_files
(self, loom_data_directory)
geneformer_001/geneformer/tokenizer.py:114
↓ 1 callersMethodvalidate_options
(self)
geneformer_001/geneformer/in_silico_perturber_stats.py:491
↓ 1 callersMethodvalidate_options
(self)
geneformer_001/geneformer/in_silico_perturber.py:774
↓ 1 callersMethodvalidate_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