MCPcopy Index your code

hub / github.com/bytedance/dplm / functions

Functions1,008 in github.com/bytedance/dplm

↓ 69 callersMethodmean
(self)
src/byprot/models/structok/modules/folding_utils/categorical_mixture.py:47
↓ 60 callersMethodlog
( self, name: str, value: _METRIC_COLLECTION, prog_bar: bool = False,
src/byprot/tasks/__init__.py:137
↓ 41 callersMethodupdate
(self, name, value, type="mean", **kwds)
src/byprot/tasks/__init__.py:270
↓ 39 callersFunctionPrintErrorAndQuit
analysis/TMalign.cpp:245
↓ 37 callersMethodclose
analysis/TMscore.cpp:1632
↓ 31 callersFunctionPrintErrorAndQuit
analysis/TMscore.cpp:2462
↓ 30 callersMethodopen
analysis/TMscore.cpp:1307
↓ 25 callersMethodwrite
analysis/TMscore.cpp:2146
↓ 24 callersMethodcompute
(self, name)
src/byprot/tasks/__init__.py:280
↓ 21 callersMethodfrom_pretrained
(cls, net_name, cfg_override={}, net_override={})
src/byprot/models/dplm/dplm_invfold.py:67
↓ 18 callersFunctiondist
analysis/TMalign.cpp:873
↓ 18 callersMethodread
analysis/TMscore.cpp:2169
↓ 16 callersFunctionDeleteArray
analysis/TMscore.cpp:2479
↓ 16 callersFunctionNewArray
analysis/TMscore.cpp:2473
↓ 16 callersMethodreset
(self, name)
src/byprot/tasks/__init__.py:283
↓ 13 callersFunction_load_model
(model_name)
src/byprot/models/structok/modules/folding_utils/pretrained.py:16
↓ 11 callersFunctionDeleteArray
analysis/TMalign.cpp:262
↓ 11 callersFunctionKabsch
Implemetation of Kabsch algoritm for finding the best rotation matrix --------------------------------------------------------------------------- x
analysis/TMalign.cpp:983
↓ 11 callersFunctionNewArray
analysis/TMalign.cpp:256
↓ 11 callersMethodget_non_special_symbol_mask
(self, output_tokens, partial_masks=None)
src/byprot/models/dplm2/dplm2.py:554
↓ 10 callersFunctionKabsch
Implemetation of Kabsch algoritm for finding the best rotation matrix --------------------------------------------------------------------------- x
analysis/TMscore.cpp:3162
↓ 10 callersFunctioncopy_t_u
copy the value of t and u into t0,u0 */
analysis/TMalign.cpp:3858
↓ 10 callersFunctiondo_rotation
analysis/TMscore.cpp:3042
↓ 10 callersMethodget_modality_type
(self, input_ids)
src/byprot/models/dplm2/dplm2.py:220
↓ 10 callersFunctiontransform
analysis/TMalign.cpp:887
↓ 9 callersFunctiongetmin
analysis/TMalign.cpp:251
↓ 9 callersFunctionmasked_mean
takes the mean of the elements of x that are not masked the mean is taken along the shared leading dims of m equivalent to: x[m].mean(tuple(ra
src/byprot/models/structok/modules/lfq.py:79
↓ 8 callersMethoddo_open
analysis/TMscore.cpp:2305
↓ 8 callersMethoderror
analysis/TMscore.cpp:1889
↓ 7 callersFunctionTMscore8_search
analysis/TMalign.cpp:1807
↓ 7 callersFunctionclean_up_after_approx_TM
analysis/TMalign.cpp:3902
↓ 7 callersFunctioncollate_dense_tensors
Takes a list of tensors with the following dimensions: [(d_11, ..., d_1K), (d_21, ..., d_2K), ..., (d_N1,
src/byprot/models/structok/modules/folding_utils/misc.py:159
↓ 7 callersFunctiondist
analysis/TMscore.cpp:3021
↓ 7 callersMethodgenerate
( self, input_tokens, max_iter=None, temperature=None, partial_masks=N
src/byprot/models/dplm2/dplm2_bit.py:343
↓ 7 callersMethodget_non_special_symbol_mask
(self, output_tokens, partial_masks=None)
src/byprot/models/dplm/dplm.py:377
↓ 6 callersFunctionDP_iter
heuristic run of dynamic programing iteratively to find the best alignment input: initial rotation matrix t, u vectors x and y, d0 output: best alignm
analysis/TMalign.cpp:2979
↓ 6 callersMethod__init__
Args: c_in: Input channel dimension c_hidden: Hidden channel dimension no
src/byprot/models/structok/modules/folding_utils/structure_module.py:90
↓ 6 callersMethod__init__
(self, config, dropout=0.1, vocab_size=None)
src/byprot/models/dplm2/modules/dplm2_modeling_esm.py:639
↓ 6 callersMethod__init__
(self, config, dropout=0.1, codebook_embed_dim=13)
src/byprot/models/dplm2/modules/dplm2_bit_modeling_esm.py:639
↓ 6 callersFunctionapprox_TM
calculate approximate TM-score given rotation matrix */
analysis/TMalign.cpp:3869
↓ 6 callersFunctiondo_rotation
analysis/TMalign.cpp:894
↓ 6 callersFunctionget_score_fast
compute the score quickly in three iterations
analysis/TMalign.cpp:2194
↓ 6 callersFunctionlabel_smoothed_nll_loss
( lprobs, target, epsilon, ignore_index=None, reduce=True )
src/byprot/modules/cross_entropy.py:10
↓ 6 callersFunctionmerge_config
(default_cfg, override_cfg)
src/byprot/utils/config.py:46
↓ 6 callersMethodread_err
analysis/TMscore.cpp:1934
↓ 6 callersFunctionrecursive_to
(obj, device)
src/byprot/utils/__init__.py:288
↓ 6 callersFunctiontopk_masking
scores: [b, n] cutoff_len: [b, 1] stochastic: bool, whether to add noise to select top_k or not returns: mask: [b, n], with 1
src/byprot/models/utils.py:257
↓ 6 callersFunctiontransform
analysis/TMscore.cpp:3035
↓ 5 callersFunctionclose_fd_array
analysis/TMscore.cpp:1372
↓ 5 callersFunctiondetailed_search
Comprehensive TMscore search engine input: two vector sets: x, y an alignment invmap0[] between x and y simplify_step: 1 or 40 or other integers sco
analysis/TMalign.cpp:2122
↓ 5 callersFunctionexists
(v)
src/byprot/models/structok/modules/lfq.py:40
↓ 5 callersMethodget_codebook_entry
(self, x, shape=None)
src/byprot/models/structok/modules/lfq.py:218
↓ 5 callersFunctionload_from_pdb
(pdb_path, process_chain=PdbDataset.process_chain)
src/byprot/utils/protein/evaluator_dplm2.py:53
↓ 5 callersFunctionload_yaml_config
(fpath: str)
src/byprot/utils/config.py:52
↓ 5 callersMethodoutput_to_pdb
(self, decoder_out, output_dir, is_trajectory=False)
src/byprot/models/structok/structok_lfq.py:268
↓ 5 callersFunctionparameter_set4final
analysis/TMalign.cpp:1687
↓ 5 callersFunctionprocess_chain
Convert a PDB chain object into a AlphaFold Protein instance. Forked from alphafold.common.protein.from_pdb_string WARNING: All non-standard
src/byprot/datamodules/pdb_dataset/utils.py:766
↓ 5 callersFunctionsample_from_categorical
(logits=None, temperature=1.0)
src/byprot/models/dplm/modules/dplm_modeling_esm.py:504
↓ 4 callersFunctionNWDP_TM
Input: score[1:len1, 1:len2], and gap_open * Output: j2i[1:len2] \in {1:len1} U {-1} * path[0:len1, 0:len2]=1,2,3, from diagonal, horizontal, vert
analysis/TMalign.cpp:1321
↓ 4 callersFunctionTMalign_main
Entry function for TM-align. Return TM-score calculation status: * 0 - full TM-score calculation * 1 - terminated due to exception * 2-7 -
analysis/TMalign.cpp:3923
↓ 4 callersFunctionTMscore8_search
analysis/TMscore.cpp:4277
↓ 4 callersMethod__init__
(self, num_tokens, codebook_dim, decay=0.99, eps=1e-5)
src/byprot/models/structok/modules/vqvae.py:429
↓ 4 callersMethod__init__
(self, r: float, batch_dim: T.Union[int, T.List[int]])
src/byprot/models/structok/modules/folding_utils/misc.py:258
↓ 4 callersMethod__init__
(self, config, dropout=0.1)
src/byprot/models/dplm/modules/dplm_modeling_esm.py:318
↓ 4 callersFunction_make_rigid_transformation_4x4
Create a rigid 4x4 transformation matrix from two axes and transl.
src/byprot/utils/protein/residue_constants.py:1105
↓ 4 callersFunction_make_rigid_transformation_4x4
Create a rigid 4x4 transformation matrix from two axes and transl.
src/byprot/datamodules/pdb_dataset/residue_constants.py:1105
↓ 4 callersFunction_maybe_remove_leading_dim
(arr: np.ndarray)
src/byprot/datamodules/pdb_dataset/protein.py:296
↓ 4 callersFunctionclose_fd
* @brief Helper function to close a file descriptor. * * Inspects @a fd and calls <b>close</b>(3) if it has a non-negative value. * *
analysis/TMscore.cpp:1354
↓ 4 callersFunctioncompute_metrics
(inference_dir)
src/byprot/utils/protein/evaluator_dplm2.py:1131
↓ 4 callersMethoddecode
x: ... NH where NH is number of codebook heads A longtensor of codebook indices, containing values from 0
src/byprot/models/structok/modules/lfq.py:257
↓ 4 callersMethoddetokenize
(self, struct_tokens, res_mask=None, **kwargs)
src/byprot/models/structok/structok_lfq.py:225
↓ 4 callersMethoddevice
(self)
src/byprot/models/dplm2/dplm2.py:190
↓ 4 callersMethodforward
(self, input_ids, **kwargs)
src/byprot/models/dplm2/dplm2_bit.py:87
↓ 4 callersMethodforward
(self, input_ids, **kwargs)
src/byprot/models/dplm2/dplm2.py:229
↓ 4 callersMethodget_non_special_sym_mask
(self, output_tokens, partial_masks=None)
src/byprot/models/dplm/dplm_invfold.py:173
↓ 4 callersMethodinfer
Runs a forward pass given input sequences. Args: sequences (Union[str, List[str]]): A list of sequences to make predictions for.
src/byprot/models/structok/modules/folding_utils/esmfold.py:274
↓ 4 callersFunctionparameter_set4final
analysis/TMscore.cpp:3119
↓ 4 callersMethodrun_evaluation
(self, batch, eval_dir)
src/byprot/utils/protein/evaluator_dplm2.py:555
↓ 4 callersMethodsample_from_logits
( self, aatype_logits, struct_logits, temperature=1.0 )
src/byprot/models/dplm2/dplm2_bit.py:321
↓ 4 callersFunctionsave_fasta
( save_name, output_results, struct_tokens=False, headers=None, continue_write=False, )
generate_dplm2.py:331
↓ 4 callersFunctionscore_fun8
1, collect those residues with dis<d; 2, calculate TMscore
analysis/TMscore.cpp:4189
↓ 4 callersFunctionscore_fun8_standard
analysis/TMscore.cpp:4232
↓ 4 callersFunctionseed_everything
Function that sets seed for pseudo-random number generators in: pytorch, numpy, python.random In addition, sets the following environment variable
src/byprot/utils/__init__.py:370
↓ 4 callersFunctionsoftmax_cross_entropy
(logits, labels)
src/byprot/models/structok/modules/loss.py:129
↓ 4 callersFunctionsplit
split a long string into vectors by whitespace * line - input string * line_vec - output vector * delimiter - delimiter */
analysis/TMscore.cpp:2554
↓ 4 callersFunctionstochastic_sample_from_categorical
( logits=None, temperature=1.0, noise_scale=1.0 )
src/byprot/models/utils.py:335
↓ 4 callersFunctionunpack_one
(t, ps, pattern)
src/byprot/models/structok/modules/lfq.py:55
↓ 4 callersMethodwait
analysis/TMscore.cpp:1757
↓ 3 callersFunctionTMscore8_search_standard
analysis/TMscore.cpp:4432
↓ 3 callersFunctionTrim
strip white space at the begining or end of string */
analysis/TMalign.cpp:354
↓ 3 callersFunctionTrim
strip white space at the begining or end of string */
analysis/TMscore.cpp:2575
↓ 3 callersMethod__init__
(self, cfg, config)
src/byprot/models/dplm/modules/dplm_adapter.py:190
↓ 3 callersMethod_build_batch_sampler
( self, dataset, max_tokens, shuffle=False, distributed=True )
src/byprot/datamodules/cath_datamodule.py:92
↓ 3 callersFunction_calc_bb_rmsd
(mask, sample_bb_pos, folded_bb_pos)
src/byprot/utils/protein/utils.py:661
↓ 3 callersMethod_init_metadata
Initialize metadata.
src/byprot/datamodules/pdb_dataset/pdb_datamodule.py:215
↓ 3 callersFunction_process_chain_id
(x)
src/byprot/datamodules/pdb_dataset/utils.py:748
↓ 3 callersFunctioncal_index_acc
(logits, target, loss_mask, bit_level=False)
src/byprot/tasks/lm/dplm2.py:22
↓ 3 callersFunctioncompute_backbone
(bb_rigids, psi_torsions)
src/byprot/utils/protein/all_atom.py:180
↓ 3 callersFunctioncompute_backbone
(bb_rigids, psi_torsions)
src/byprot/datamodules/pdb_dataset/all_atom.py:180
↓ 3 callersFunctioncompute_fape
Computes FAPE loss. Args: pred_frames: [*, N_frames] Rigid object of predicted frames target_frames: [*,
src/byprot/models/structok/modules/loss.py:174
next →1–100 of 1,008, ranked by callers