deepmd.dpmodel.utils.neighbor_stat

Module Contents

Classes

NeighborStatOP	Class for getting neighbor statics data information.
NeighborStat	Neighbor statistics using pure NumPy.

class deepmd.dpmodel.utils.neighbor_stat.NeighborStatOP(ntypes: int, rcut: float, mixed_types: bool)

Bases: deepmd.dpmodel.common.NativeOP

Class for getting neighbor statics data information.

Parameters:

ntypes

The num of atom types

rcut

The cut-off radius

mixed_typesbool, optional

If True, treat all types as a single type.

call(coord: numpy.ndarray, atype: numpy.ndarray, cell: numpy.ndarray | None) → Tuple[float, numpy.ndarray]

Calculate the neareest neighbor distance between atoms, maximum nbor size of atoms and the output data range of the environment matrix.

Parameters:

coord

The coordinates of atoms.

atype

The atom types.

cell

The cell.

Returns:

float

The minimal squared distance between two atoms

np.ndarray

The maximal number of neighbors

class deepmd.dpmodel.utils.neighbor_stat.NeighborStat(ntypes: int, rcut: float, mixed_type: bool = False)

Bases: deepmd.utils.neighbor_stat.NeighborStat

Neighbor statistics using pure NumPy.

Parameters:

ntypesint

The num of atom types

rcutfloat

The cut-off radius

mixed_typebool, optional, default=False

Treat all types as a single type.

iterator(data: deepmd.utils.data_system.DeepmdDataSystem) → Iterator[Tuple[numpy.ndarray, float, str]]

Abstract method for producing data.

Yields:

np.ndarray

The maximal number of neighbors

float

The squared minimal distance between two atoms

str

The directory of the data system