deepmd.tf.utils.weight_avg

Alias for backward compatibility.

Functions

weighted_average(→ dict)

Compute weighted average of prediction errors (MAE or RMSE) for model.

Module Contents

deepmd.tf.utils.weight_avg.weighted_average(errors: list[dict[str, tuple[float, float]]]) → dict

Compute weighted average of prediction errors (MAE or RMSE) for model.

Parameters:

errorslist[dict[str, tuple[float, float]]]

List: the error of systems Dict: the error of quantities, name given by the key str: the name of the quantity, must starts with ‘mae’ or ‘rmse’ Tuple: (error, weight)

Returns:

Dict

weighted averages