deepmd.pt.model.atomic_model.dos_atomic_model
Module Contents
Classes
Model give atomic prediction of some physical property. |
- class deepmd.pt.model.atomic_model.dos_atomic_model.DPDOSAtomicModel(descriptor, fitting, type_map, **kwargs)[source]
Bases:
deepmd.pt.model.atomic_model.dp_atomic_model.DPAtomicModelModel give atomic prediction of some physical property.
- Parameters:
- descriptor
Descriptor
- fitting_net
Fitting net
- type_map
Mapping atom type to the name (str) of the type. For example type_map[1] gives the name of the type 1.