deepmd.tf.model.model_stat

Alias for backward compatibility.

Functions

_make_all_stat_ref(→ dict[str, list[Any]])
make_stat_input(→ dict[str, list[Any]])	Collect batches from a DeepmdDataSystem into a dict of lists.
merge_sys_stat(→ dict[str, list[Any]])

Module Contents

deepmd.tf.model.model_stat._make_all_stat_ref(data: Any, nbatches: int) → dict[str, list[Any]]

deepmd.tf.model.model_stat.make_stat_input(data: Any, nbatches: int, merge_sys: bool = True) → dict[str, list[Any]]

Collect batches from a DeepmdDataSystem into a dict of lists.

This is a low-level helper used by the TF backend and by make_stat_input().

Parameters:

data

The data (must support get_nsystems() and get_batch(sys_idx=))

nbatchesint

The number of batches per system

merge_sysbool (True)

Merge system data

Returns:

all_stat:

A dictionary of list of list storing data for stat. if merge_sys == False data can be accessed by

all_stat[key][sys_idx][batch_idx][frame_idx]

else merge_sys == True can be accessed by

all_stat[key][batch_idx][frame_idx]

deepmd.tf.model.model_stat.merge_sys_stat(all_stat: dict[str, list[Any]]) → dict[str, list[Any]]