prxteinmpnn.mpnn

Core module for the PrxteinMPNN model.

prxteinmpnn.mpnn

class prxteinmpnn.mpnn.ProteinMPNNModelVersion(value)

Bases: Enum

Enum for different ProteinMPNN model configurations.

V_48_002 = 'v_48_002.pkl'

V_48_010 = 'v_48_010.pkl'

V_48_020 = 'v_48_020.pkl'

V_48_030 = 'v_48_030.pkl'

class prxteinmpnn.mpnn.ModelWeights(value)

Bases: Enum

Enum for different sets of model weights.

DEFAULT = 'original'

SOLUBLE = 'soluble'

prxteinmpnn.mpnn.get_mpnn_model(model_version=ProteinMPNNModelVersion.V_48_020, model_weights=ModelWeights.DEFAULT)

Create a ProteinMPNN model with specified configuration and weights.

Parameters:

• model_version (ProteinMPNNModelVersion) – The model configuration to use.

• model_weights (ModelWeights) – 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()