deepmd.tf.utils.weight_avg

Alias for backward compatibility.

Functions

weighted_average(→ Dict)

Compute wighted 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 wighted 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