megnet.utils.models module

Model utilities, mainly for model loading and download

load_model(model_name: str) → megnet.models.base.GraphModel

load the model by user friendly name as in megnet.utils.models.AVAILABEL_MODELS

Parameters

model_name – str model name string

Returns: GraphModel