MCPcopy Create free account

hub / github.com/bytedance/dplm / functions

Functions1,008 in github.com/bytedance/dplm

↓ 1 callersFunctioncalc_tm_score
(pos_1, pos_2, seq_1, seq_2, mask)
src/byprot/modules/protein_metrics.py:59
↓ 1 callersFunctioncalc_tm_score
(pos_1, pos_2, seq_1, seq_2)
src/byprot/utils/protein/utils.py:900
↓ 1 callersFunctioncalculate_score_gotoh
calculate dynamic programming matrix using gotoh algorithm * S - cumulative scorefor each cell * P - string representation for path *
analysis/TMscore.cpp:3716
↓ 1 callersMethodcluster_size_ema_update
(self, new_cluster_size)
src/byprot/models/structok/modules/vqvae.py:444
↓ 1 callersFunctioncollate
(tokenizer, init_aa_seq, init_struct_seq, args, device)
src/byprot/utils/scaffold_utils.py:511
↓ 1 callersFunctioncollate_fn
(batch: list)
src/byprot/datamodules/pdb_dataset/pdb_datamodule.py:51
↓ 1 callersFunctioncompose_config
(**kwds)
src/byprot/utils/config.py:42
↓ 1 callersMethodcompute_forward_folding_metrics
(self, output_dir)
src/byprot/utils/protein/evaluator_dplm2.py:1074
↓ 1 callersMethodcompute_inverse_folding_metrics
(self, output_dir)
src/byprot/utils/protein/evaluator_dplm2.py:1095
↓ 1 callersFunctioncompute_predicted_aligned_error
Computes aligned confidence metrics from logits. Args: logits: [*, num_res, num_res, num_bins] the logits output from PredictedAlig
src/byprot/models/structok/modules/loss.py:680
↓ 1 callersMethodcompute_reconstruction_metrics
(self, output_dir)
src/byprot/utils/protein/evaluator_dplm2.py:1051
↓ 1 callersMethodcompute_sample_metrics
( self, batch, model_traj, bb_traj, aa_traj, clean_aa_traj,
src/byprot/utils/protein/evaluator_dplm2.py:830
↓ 1 callersFunctioncompute_tm
( logits: torch.Tensor, residue_weights: Optional[torch.Tensor] = None, max_bin: int = 31, no_
src/byprot/models/structok/modules/loss.py:720
↓ 1 callersMethodcompute_unconditional_metrics
(self, output_dir)
src/byprot/utils/protein/evaluator_dplm2.py:1016
↓ 1 callersFunctioncompute_violation_metrics
Compute several metrics to assess the structural violations.
src/byprot/models/structok/modules/loss.py:1381
↓ 1 callersFunctionconditional_generate_from_fasta
(args)
generate_dplm2.py:291
↓ 1 callersMethodcopy_to
(self, model)
src/byprot/models/structok/modules/ema.py:62
↓ 1 callersFunctioncreate_batched_sequence_datasest
( sequences: T.List[T.Tuple[str, str]], max_tokens_per_batch: int = 1024 )
analysis/cal_plddt_dir.py:126
↓ 1 callersFunctioncreate_batches
(batch, args)
src/byprot/utils/scaffold_utils.py:599
↓ 1 callersFunctioncreate_idxs_list
(pdb, tokenizer, batch, args)
src/byprot/utils/scaffold_utils.py:567
↓ 1 callersFunctioncreate_init_seq
(length)
generate_dplm2.py:139
↓ 1 callersFunctioncreate_init_seq
(pdb, ori_pdb, aa_seq, struct_seq, tokenizer, args)
src/byprot/utils/scaffold_utils.py:410
↓ 1 callersFunctioncreate_parser
()
analysis/cal_plddt_dir.py:142
↓ 1 callersMethoddecode
(self, batch_ids, return_as="str", remove_special=False)
src/byprot/datamodules/dataset/data_utils.py:108
↓ 1 callersMethoddevice
(self)
src/byprot/models/structok/modules/folding_utils/esmfold.py:354
↓ 1 callersFunctiondistance
(p, eps=1e-10)
src/byprot/utils/protein/utils.py:572
↓ 1 callersMethoddistogram
(coords, min_bin, max_bin, num_bins)
src/byprot/models/structok/modules/folding_utils/trunk.py:296
↓ 1 callersFunctiondistogram_loss
( logits, pseudo_beta, pseudo_beta_mask, min_bin=2.3125, max_bin=21.6875, no_bins=64,
src/byprot/models/structok/modules/loss.py:612
↓ 1 callersMethodembed_avg_ema_update
(self, new_embed_avg)
src/byprot/models/structok/modules/vqvae.py:449
↓ 1 callersFunctionenable_cpu_offloading
(model)
analysis/cal_plddt_dir.py:95
↓ 1 callersFunctionencode_sequence
( seq: str, residue_index_offset: T.Optional[int] = 512, chain_linker: T.Optional[str] = "G" * 25,
src/byprot/models/structok/modules/folding_utils/misc.py:30
↓ 1 callersFunctionentropy_loss
Entropy loss of unnormalized logits. logits: Affinities are over the last dimension https://github.com/google-research/magvit/blob/05e8cfd65
src/byprot/models/structok/modules/lfq.py:95
↓ 1 callersMethodeval_self_consistency
(self, pred_ids, positions, mask=None)
src/byprot/tasks/lm/dplm_invfold.py:337
↓ 1 callersMethodevaluate_forward_folding
(self, pdb_folder, inplace_save=False)
src/byprot/utils/protein/evaluator_dplm2.py:448
↓ 1 callersMethodevaluate_inverse_folding
(self, fasta_path, inplace_save=False)
src/byprot/utils/protein/evaluator_dplm2.py:489
↓ 1 callersMethodevaluate_reconstruction
(self, pdb_folder, inplace_save=False)
src/byprot/utils/protein/evaluator_dplm2.py:364
↓ 1 callersMethodevaluate_unconditional
(self, pdb_folder, inplace_save=False)
src/byprot/utils/protein/evaluator_dplm2.py:419
↓ 1 callersFunctionexists
(obj)
src/byprot/models/dplm2/dplm2.py:20
↓ 1 callersMethodexited
analysis/TMscore.cpp:1866
↓ 1 callersFunctionextract_aln_from_resi
extract pairwise sequence alignment from residue index vectors, * assuming that "sequence" contains two empty strings. * return length of alignmen
analysis/TMalign.cpp:788
↓ 1 callersFunctionextract_aln_from_resi
extract pairwise sequence alignment from residue index vectors, * assuming that "sequence" contains two empty strings. * return length of alignmen
analysis/TMscore.cpp:4053
↓ 1 callersFunctionextract_coords_from_structure
Args: structure: An instance of biotite AtomArray atoms: default ["N", "CA", "C"] Returns: Tuple (coords, seq)
src/byprot/utils/io.py:88
↓ 1 callersFunctionextreme_ca_ca_distance_violations
Counts residues whose Ca is a large distance from its neighbour. Measures the fraction of CA-CA pairs between consecutive amino acids that are
src/byprot/models/structok/modules/loss.py:1344
↓ 1 callersFunctionfape_loss
( out: Dict[str, torch.Tensor], batch: Dict[str, torch.Tensor], config: ml_collections.ConfigDict,
src/byprot/models/structok/modules/loss.py:355
↓ 1 callersFunctionfilter_backbone2
Filter all peptide backbone atoms of one array. This includes the "N", "CA" and "C" atoms of amino acids. Parameters ---------- arra
src/byprot/utils/io.py:28
↓ 1 callersFunctionfind_highest_align_score
locate the cell with highest alignment score. reset path after * the cell to zero */
analysis/TMscore.cpp:3669
↓ 1 callersFunctionfloat_fmt
(float_value)
src/byprot/utils/callbacks.py:88
↓ 1 callersFunctionformat_check
(args)
generate_dplm.py:11
↓ 1 callersMethodforward
(self, batch, return_ids=False)
src/byprot/tasks/lm/mlm.py:277
↓ 1 callersMethodforward
(self, batch, return_ids=False)
src/byprot/tasks/lm/dplm_invfold.py:427
↓ 1 callersMethodforward
Runs a forward pass given input tokens. Use `model.infer` to run inference from a sequence. Args: aa (torch.Tensor): Tens
src/byprot/models/structok/modules/folding_utils/decoder.py:185
↓ 1 callersMethodforward
Runs a forward pass given input tokens. Use `model.infer` to run inference from a sequence. Args: aa (torch.Tensor): Tens
src/byprot/models/structok/modules/folding_utils/esmfold.py:147
↓ 1 callersMethodforward
(self, input_ids, return_last_hidden_state=False, **kwargs)
src/byprot/models/dplm/dplm.py:150
↓ 1 callersMethodforward
( self, input_ids, attention_mask=None, inputs_embeds=None, decoder_in
src/byprot/models/dplm/modules/dplm_modeling_esm.py:337
↓ 1 callersMethodforward
( self, batch, encoder_out=None, tokens=None, loss_mask=None,
src/byprot/models/dplm/modules/dplm_adapter.py:72
↓ 1 callersMethodforward_decoder
( self, prev_decoder_out, need_attn_weights=False, partial_masks=None,
src/byprot/models/dplm2/dplm2_bit.py:254
↓ 1 callersMethodforward_decoder
( self, prev_decoder_out, need_attn_weights=False, partial_masks=None,
src/byprot/models/dplm2/dplm2.py:477
↓ 1 callersMethodforward_decoder
( self, prev_decoder_out, encoder_out=None, need_attn_weights=False, p
src/byprot/models/dplm/dplm.py:304
↓ 1 callersMethodforward_decoder
( self, prev_decoder_out, encoder_out, need_attn_weights=False, partia
src/byprot/models/dplm/dplm_invfold.py:183
↓ 1 callersMethodforward_decoder
( self, prev_decoder_out, encoder_out, need_attn_weights=False, partia
src/byprot/models/dplm/modules/dplm_modeling_esm.py:397
↓ 1 callersMethodforward_encoder
(self, input_tokens, **kwargs)
src/byprot/models/dplm2/dplm2.py:456
↓ 1 callersMethodforward_encoder
(self, batch, **kwargs)
src/byprot/models/dplm2/modules/dplm2_modeling_esm.py:688
↓ 1 callersMethodforward_encoder
(self, input_tokens, **kwargs)
src/byprot/models/dplm/dplm.py:205
↓ 1 callersMethodforward_encoder
(self, batch, use_draft_seq=False)
src/byprot/models/dplm/dplm_invfold.py:147
↓ 1 callersMethodforward_encoder
(self, batch, **kwargs)
src/byprot/models/dplm/modules/dplm_modeling_esm.py:368
↓ 1 callersMethodframes_and_literature_positions_to_atom14_pos
( self, r, f )
src/byprot/models/structok/modules/folding_utils/structure_module.py:852
↓ 1 callersFunctionframes_to_atom14_pos
Convert frames to their idealized all atom representation. Args: r: All rigid groups. [..., N, 8, 3] aatype: Residue types. [...,
src/byprot/utils/protein/all_atom.py:144
↓ 1 callersFunctionframes_to_atom14_pos
Convert frames to their idealized all atom representation. Args: r: All rigid groups. [..., N, 8, 3] aatype: Residue types. [...,
src/byprot/datamodules/pdb_dataset/all_atom.py:144
↓ 1 callersMethodfrom_pretrained
(cls, cfg)
src/byprot/models/dplm/modules/dplm_adapter.py:41
↓ 1 callersFunctiongenerate
(args)
generate_dplm.py:73
↓ 1 callersFunctiongenerate
(args, saveto)
run/scaffold_generate_dplm2.py:17
↓ 1 callersFunctiongenerate
(args, saveto)
run/scaffold_generate_dplm.py:19
↓ 1 callersFunctionget_atom_coords_residuewise
Example for atoms argument: ["N", "CA", "C"]
src/byprot/utils/io.py:123
↓ 1 callersMethodget_featurizer
(self, name="cath", **kwds)
src/byprot/datamodules/dataset/data_utils.py:87
↓ 1 callersFunctionget_git_revision_hash
()
src/byprot/utils/__init__.py:359
↓ 1 callersFunctionget_initial
perform gapless threading to find the best initial alignment input: x, y, xlen, ylen output: y2x0 stores the best alignment: e.g., y2x0[j]=i means: th
analysis/TMalign.cpp:2341
↓ 1 callersFunctionget_initial5
get_initial5 in TMalign fortran, get_initial_local in TMalign c by yangji get initial alignment of local structure superposition input: x, y, xlen, yl
analysis/TMalign.cpp:2514
↓ 1 callersFunctionget_initial_dplm
(args, tokenizer, pdb, ori_pdb, device)
src/byprot/utils/scaffold_utils.py:300
↓ 1 callersFunctionget_initial_dplm2
(args, aa_seq, struct_seq, tokenizer, pdb, ori_pdb, device)
src/byprot/utils/scaffold_utils.py:395
↓ 1 callersFunctionget_initial_fgt
perform fragment gapless threading to find the best initial alignment input: x, y, xlen, ylen output: y2x0 stores the best alignment: e.g., y2x0[j]=i
analysis/TMalign.cpp:2744
↓ 1 callersFunctionget_initial_ss
get initial alignment from secondary structure alignment input: x, y, xlen, ylen output: y2x stores the best alignment: e.g., y2x[j]=i means: the jth
analysis/TMalign.cpp:2499
↓ 1 callersFunctionget_initial_ssplus
get initial alignment from secondary structure and previous alignments input: x, y, xlen, ylen output: y2x stores the best alignment: e.g., y2x[j]=i m
analysis/TMalign.cpp:2665
↓ 1 callersFunctionget_logger
Initializes multi-GPU-friendly python command line logger.
src/byprot/utils/__init__.py:38
↓ 1 callersMethodget_metadata_lens
(self)
src/byprot/datamodules/dataset/uniref.py:214
↓ 1 callersMethodget_metadata_lens
(self)
src/byprot/datamodules/dataset/tokenized_protein.py:279
↓ 1 callersMethodget_metadata_lens
(self)
src/byprot/datamodules/dataset/uniref_hf.py:197
↓ 1 callersFunctionget_net_dplm2
(cfg)
src/byprot/models/utils.py:109
↓ 1 callersFunctionget_net_dplm2_bit
(cfg)
src/byprot/models/utils.py:209
↓ 1 callersMethodget_non_special_sym_mask
(self, output_tokens, partial_masks=None)
src/byprot/models/dplm/modules/dplm_adapter.py:178
↓ 1 callersFunctionget_obj_from_str
(string, reload=False)
src/byprot/utils/config.py:109
↓ 1 callersFunctiongetmin
analysis/TMscore.cpp:2468
↓ 1 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/decoder.py:337
↓ 1 callersMethodinfer_pdbs
Returns list of pdb (files) strings from the model given a list of input sequences.
src/byprot/models/structok/modules/folding_utils/decoder.py:408
↓ 1 callersMethodinfer_pdbs
Returns list of pdb (files) strings from the model given a list of input sequences.
src/byprot/models/structok/modules/folding_utils/esmfold.py:336
↓ 1 callersFunctioninit_gotoh_mat
initialize matrix in gotoh algorithm */
analysis/TMscore.cpp:3632
↓ 1 callersFunctioninit_model_on_gpu_with_cpu_offloading
(model)
analysis/cal_plddt_dir.py:117
↓ 1 callersFunctioninitialize_conditional_generation
( fasta_path, tokenizer, device, args, model=None )
generate_dplm2.py:16
↓ 1 callersFunctioninitialize_generation
( num_seqs, length, tokenizer, device, cond_seq=None, args=None )
generate_dplm.py:42
↓ 1 callersFunctioninitialize_generation
( task, num_seqs, length, tokenizer, device, batch_size=50 )
generate_dplm2.py:136
↓ 1 callersMethodinitialize_output_tokens
(self, input_tokens, partial_masks=None, **kwargs)
src/byprot/models/dplm/dplm.py:208
← previousnext →301–400 of 1,008, ranked by callers