deepfold.common.protein.from_prediction

deepfold.common.protein.from_prediction(processed_features: Mapping[str, ndarray], result: Mapping[str, Any], b_factors: ndarray | None = None, remove_leading_feature_dimension: bool = False, is_trajectory: bool = False, remark: str | None = None, parents: Sequence[str] | None = None, parents_chain_index: Sequence[int] | None = None) → Protein | List[Protein][source]

Assembles a protein from a prediction.

Parameters:

  • processed_features – Dictionary holding model inputs.

  • result – Dictionary holding model outputs.

  • b_factors – (Optional) B-factors to use for the protein.

  • remove_leading_feature_dimension – Whether to remove the leading dimension of the feature values.

  • remark – (Optional) Remark about the prediction

  • parents – (Optional) List of template names

Returns:

A protein instance.