MCPcopy Create free account

hub / github.com/apple/ml-simplefold / functions

Functions396 in github.com/apple/ml-simplefold

Method_basic_init
(module)
src/simplefold/model/torch/layers.py:250
Method_basic_init
(module)
src/simplefold/model/torch/blocks.py:46
Functionaatype_to_str_sequence
(aatype)
src/simplefold/utils/residue_constants.py:1312
Functionbootstrap_ci
Compute 95% bootstrap confidence intervals for the mean of 'series'.
src/simplefold/evaluation/analyze_folding.py:136
Methodclear
(self)
src/simplefold/evaluation/analyze_two_state.py:150
Functioncompute_all_coords_mask
(structure)
src/simplefold/boltz_data_pipeline/feature/symmetry.py:388
Methodcompute_alpha_t
Compute the data coefficient along the path
src/simplefold/model/flow.py:91
Functioncompute_covalent_ligands
Compute the covalent ligands from a list of connections. Parameters ---------- connections: List[gemmi.Connection] The connection
src/simplefold/utils/mmcif_utils.py:276
Methodcompute_d_alpha_alpha_ratio_t
Compute the ratio between d_alpha and alpha
src/simplefold/model/flow.py:99
Methodcompute_sigma_t
Compute the noise coefficient along the path
src/simplefold/model/flow.py:95
Functioncompute_symmetry_idx_dictionary
(data)
src/simplefold/boltz_data_pipeline/feature/symmetry.py:78
Methodcompute_velocity_from_score
(self, s_t, y_t, t)
src/simplefold/model/flow.py:74
Methodconfigure_optimizers
(self)
src/simplefold/model/simplefold.py:764
Methodcrop
Crop the data to a maximum number of tokens. Parameters ---------- data : Tokenized The tokenized data. m
src/simplefold/boltz_data_pipeline/crop/slice.py:64
Methodcrop
Crop the data to a maximum number of tokens. Parameters ---------- data : Tokenized The tokenized data. m
src/simplefold/boltz_data_pipeline/crop/boltz.py:161
Functiondefault
(v, d)
src/simplefold/utils/boltz_utils.py:36
Functiondefault
returns x if it exists, otherwise y
src/simplefold/model/mlx/layers.py:133
Methoddump
Dump the object to an NPZ file. Parameters ---------- path : Path The path to the file.
src/simplefold/boltz_data_pipeline/types.py:42
Functionenforce_tags
Prompts user to input tags from command line if no tags are provided in config. :param cfg: A DictConfig composed by Hydra. :param save_to_fi
src/simplefold/utils/utils.py:89
Methodfilter
Filter a chains based on their length. Parameters ---------- structure : Structure The structure to filter chains
src/simplefold/boltz_data_pipeline/filter/static/polymer.py:41
Methodfilter
Filter proteins with all unknown residues. Parameters ---------- structure : Structure The structure to filter ch
src/simplefold/boltz_data_pipeline/filter/static/polymer.py:75
Methodfilter
Filter protein if consecutive CA atoms above a threshold. Parameters ---------- structure : Structure The structu
src/simplefold/boltz_data_pipeline/filter/static/polymer.py:125
Methodfilter
Filter out clashing chains. Parameters ---------- structure : Structure The structure to filter chains from.
src/simplefold/boltz_data_pipeline/filter/static/polymer.py:209
Methodfilter
Filter chains in a structure. Parameters ---------- structure : Structure The structure to filter chains from.
src/simplefold/boltz_data_pipeline/filter/static/filter.py:20
Methodfilter
Filter excluded ligands. Parameters ---------- structure : Structure The structure to filter chains from.
src/simplefold/boltz_data_pipeline/filter/static/ligand.py:153
Methodfilter
Filter a data record. Parameters ---------- record : Record The object to consider filtering in / out. R
src/simplefold/boltz_data_pipeline/filter/dynamic/subset.py:33
Methodfilter
Filter a data record. Parameters ---------- record : Record The object to consider filtering in / out. R
src/simplefold/boltz_data_pipeline/filter/dynamic/filter.py:18
Methodfilter
Filter complexes based on their resolution. Parameters ---------- record : Record The record to filter.
src/simplefold/boltz_data_pipeline/filter/dynamic/resolution.py:27
Methodfilter
Filter structures based on their resolution. Parameters ---------- record : Record The record to filter.
src/simplefold/boltz_data_pipeline/filter/dynamic/max_residues.py:30
Methodfilter
Filter a record based on its date. Parameters ---------- record : Record The record to filter. Returns
src/simplefold/boltz_data_pipeline/filter/dynamic/date.py:51
Methodforward
(self, x, **kwargs)
src/simplefold/model/torch/layers.py:51
Methodforward
(self, x, **kwargs)
src/simplefold/model/torch/layers.py:88
Methodforward
(self, x)
src/simplefold/model/torch/layers.py:141
Methodforward
(self, t)
src/simplefold/model/torch/layers.py:194
Methodforward
(self, cond, train, force_drop_ids=None)
src/simplefold/model/torch/layers.py:226
Methodforward
(self, x, c)
src/simplefold/model/torch/layers.py:264
Methodforward
(self, x)
src/simplefold/model/torch/layers.py:295
Methodforward
( self, latents, c, **kwargs, )
src/simplefold/model/torch/blocks.py:58
Methodforward
( self, latents, **kwargs, )
src/simplefold/model/torch/blocks.py:105
Methodforward
(self, latents, c, **kwargs)
src/simplefold/model/torch/blocks.py:121
Methodforward
(self, noised_pos, t, feats, self_cond=None)
src/simplefold/model/torch/architecture.py:172
Methodforward
(self, pos)
src/simplefold/model/torch/pos_embed.py:20
Methodforward
Get the positional encoding for each coordinate. Args: pos: (*, in_dim) Returns: out:
src/simplefold/model/torch/pos_embed.py:94
Methodforward
xq: [B, H, N, D] xk: [B, H, N, D] pos: [B, N, in_dim]
src/simplefold/model/torch/pos_embed.py:167
Methodforward
( self, latent, feats, )
src/simplefold/model/torch/confidence_module.py:55
Methodfrom_pretrained_folding_model
(self)
src/simplefold/wrapper.py:78
Methodfrom_pretrained_plddt_model
(self)
src/simplefold/wrapper.py:117
Methodget_atoms
(self)
src/simplefold/boltz_data_pipeline/write/mmcif.py:128
Functionget_current_idx_list
(data)
src/simplefold/boltz_data_pipeline/feature/symmetry.py:93
Functionget_symmetries
Create a dictionary for the ligand symmetries. Parameters ---------- path : str The path to the ligand symmetries. Returns
src/simplefold/boltz_data_pipeline/feature/symmetry.py:43
Functionget_unk_token
Get the unknown token for a given entity type. Parameters ---------- dtype : gemmi.EntityType The entity type. Returns -
src/simplefold/utils/mmcif_utils.py:210
Methodload
Load the object from an NPZ file. Parameters ---------- path : Path The path to the file. Returns
src/simplefold/boltz_data_pipeline/types.py:26
Methodload
Load the object from a JSON file. Parameters ---------- path : Path The path to the file. Returns
src/simplefold/boltz_data_pipeline/types.py:58
Methodload
Load a structure from an NPZ file. Parameters ---------- path : Path The path to the file. Returns
src/simplefold/boltz_data_pipeline/types.py:160
Functionlog
(t, eps=1e-20)
src/simplefold/utils/boltz_utils.py:40
Functionmain
()
src/simplefold/cli.py:14
Functionmake_atom14_dists_bounds
compute upper and lower bounds for bonds to assess violations.
src/simplefold/utils/residue_constants.py:1224
Functionmap_plddt_torch_to_mlx
(key, value)
src/simplefold/utils/mlx_utils.py:79
Functionmap_torch_to_mlx
(key, value)
src/simplefold/utils/mlx_utils.py:49
Functionminimum_lddt_symmetry_coords
( coords: torch.Tensor, feats: dict, index_batch: int, **args_rmsd, )
src/simplefold/boltz_data_pipeline/feature/symmetry.py:247
Functionminimum_symmetry_coords
( coords: torch.Tensor, feats: dict, index_batch: int, **args_rmsd, )
src/simplefold/boltz_data_pipeline/feature/symmetry.py:113
Methodon_before_optimizer_step
(self, optimizer: Optimizer)
src/simplefold/model/simplefold.py:685
Methodon_before_zero_grad
(self, optimizer: Optimizer)
src/simplefold/model/simplefold.py:710
Methodon_load_checkpoint
(self, checkpoint)
src/simplefold/model/simplefold.py:748
Methodon_predict_epoch_end
Print the number of failed examples.
src/simplefold/boltz_data_pipeline/write/writer.py:230
Methodon_predict_epoch_end
(self)
src/simplefold/model/simplefold.py:610
Methodon_save_checkpoint
(self, checkpoint)
src/simplefold/model/simplefold.py:727
Methodon_train_batch_end
(self, outputs, batch, batch_idx)
src/simplefold/model/simplefold.py:674
Methodon_train_start
(self)
src/simplefold/model/simplefold.py:507
Functionparse_a3m
Process an A3M file. Parameters ---------- path : Path The path to the a3m(.gz) file. taxonomy : Redis The taxonomy d
src/simplefold/boltz_data_pipeline/parse/a3m.py:112
Functionparse_csv
Process an A3M file. Parameters ---------- path : Path The path to the a3m(.gz) file. max_seqs : int, optional The ma
src/simplefold/boltz_data_pipeline/parse/csv.py:19
Functionparse_fasta
Parse a fasta file. The name of the fasta file is used as the name of this job. We rely on the fasta record id to determine the entity type.
src/simplefold/boltz_data_pipeline/parse/fasta.py:19
Functionparse_yaml
Parse a Boltz input yaml / json. The input file should be a yaml file with the following format: sequences: - protein: i
src/simplefold/boltz_data_pipeline/parse/yaml.py:18
Functionpick_interface_token
Pick a random token from an interface. Parameters ---------- tokens : np.ndarray The token data. interface : int The
src/simplefold/boltz_data_pipeline/crop/boltz.py:76
Methodpredict_contacts
(self, tokens)
src/simplefold/model/mlx/esm_network.py:159
Methodpredict_step
(self, batch, batch_idx)
src/simplefold/model/simplefold.py:513
Methodprocess_input
(self, aa_seq)
src/simplefold/wrapper.py:293
Methodregister
(self, name, tensor)
src/simplefold/model/simplefold.py:140
Functionrenumber_pdb
(molseq, X_path, X_renum, start=1)
src/simplefold/evaluation/analyze_two_state.py:173
Methodrun_inference
(self, batch, model, plddt_model, device)
src/simplefold/wrapper.py:321
Methodsample
(self, model_fn, flow, noise, batch)
src/simplefold/model/torch/sampler.py:83
Methodsave_result
(self, structure, record, results, out_name)
src/simplefold/wrapper.py:369
Functionselect_subset_from_mask
(mask, p)
src/simplefold/boltz_data_pipeline/feature/featurizer.py:28
Functionsequence_to_onehot
Maps the given sequence into a one-hot encoded matrix. Args: sequence: An amino acid sequence. mapping: A dictionary mapping amino ac
src/simplefold/utils/residue_constants.py:890
Methodsetup
Lightning hook that is called at the beginning of fit (train + validate), validate, test, or predict. This is a good hook when you ne
src/simplefold/model/simplefold.py:622
Methodsetup
Run the setup for the DataModule. Parameters ---------- stage : str, optional The stage, one of 'fit', 'validate'
src/simplefold/datasets/test_datamodule.py:174
Methodsetup
Run the setup for the DataModule. Parameters ---------- stage : str, optional The stage, one of 'fit', 'validate'
src/simplefold/datasets/train_datamodule.py:354
Methodstep
(self, epoch=None)
src/simplefold/utils/lr_scheduler.py:48
Functiontask_wrapper
Optional decorator that controls the failure behavior when executing the task function. This wrapper can be used to: - make sure loggers
src/simplefold/utils/utils.py:160
Functiontm_diversity
Calculate diversity among pairs of modeled protein structures for each score. Returns: dict of scores('rmsd': float, 'tm': float, 'gd
src/simplefold/evaluation/two_state_utils/tm_ens.py:84
Functiontm_ensemble_at_N
(samples, gt1, gt2, N=5)
src/simplefold/evaluation/two_state_utils/tm_ens.py:64
Methodtokenize
Tokenize the input data. Parameters ---------- data : Input The input data. Returns -------
src/simplefold/boltz_data_pipeline/tokenize/boltz_protein.py:42
Methodtraining_step
(self, batch, batch_idx)
src/simplefold/model/simplefold.py:496
Methodval_dataloader
Get the validation dataloader. Returns ------- DataLoader The validation dataloader.
src/simplefold/datasets/train_datamodule.py:392
Methodvalidation_step
(self, batch, batch_idx)
src/simplefold/model/simplefold.py:503
Methodwrite_on_batch_end
Write the predictions to disk.
src/simplefold/boltz_data_pipeline/write/writer.py:59
← previous301–396 of 396, ranked by callers