deepmd.pt.model.model.dp_model

Module Contents

Classes

DPModelCommon

A base class to implement common methods for all the Models.
class deepmd.pt.model.model.dp_model.DPModelCommon[source]

A base class to implement common methods for all the Models.

classmethod update_sel(global_jdata: dict, local_jdata: dict)[source]

Update the selection and perform neighbor statistics.

Parameters:
global_jdatadict

The global data, containing the training section

local_jdatadict

The local data refer to the current class

get_fitting_net()[source]

Get the fitting network.

get_descriptor()[source]

Get the descriptor.