prxteinmpnn.mpnn

Contents

prxteinmpnn.mpnn#

Core module for the PrxteinMPNN model.

prxteinmpnn.mpnn

prxteinmpnn.mpnn.get_mpnn_model(model_version='v_48_020.pkl', model_weights='original')[source]#

Create a ProteinMPNN model with specified configuration and weights.

Parameters:

  • model_version (Literal['v_48_002.pkl', 'v_48_010.pkl', 'v_48_020.pkl', 'v_48_030.pkl']) – The model configuration to use.

  • model_weights (Literal['original', 'soluble']) – The weights to load for the model.

Return type:

PyTree

Returns:

A PyTree containing the model parameters.

Raises:

FileNotFoundError – If the model file does not exist.

Example

>>> params = get_mpnn_model()
Contents