deepmd.utils.weight_avg

Module Contents

Functions

weighted_average(→ Dict)

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

deepmd.utils.weight_avg.weighted_average(errors: List[Dict[str, Tuple[float, float]]]) Dict[source]

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