Function RMSE

tasks/AutoPower/code/metrics.py:4–7

(predictions, targets)

import torch.nn as nn

def RMSE(predictions, targets):
    mse_eval = nn.MSELoss()
    rmse = torch.sqrt(mse_eval(predictions, targets)).item()
    return rmse