deepmd.jax.model.model

deepmd.jax.model.model#

Functions#

get_standard_model(data)

Get a Model from a dictionary.

get_zbl_model(→ deepmd.jax.model.dp_zbl_model.DPZBLModel)

get_model(data)

Get a model from a dictionary.

Module Contents#

deepmd.jax.model.model.get_standard_model(data: dict)[source]#

Get a Model from a dictionary.

Parameters:
datadict

The data to construct the model.

deepmd.jax.model.model.get_zbl_model(data: dict) → deepmd.jax.model.dp_zbl_model.DPZBLModel[source]#
deepmd.jax.model.model.get_model(data: dict)[source]#

Get a model from a dictionary.

Parameters:
datadict

The data to construct the model.