pyepo.metric.mse

Mean Squared Error

Functions

MSE(→ float)

A function to evaluate model performance with MSE

Module Contents

pyepo.metric.mse.MSE(predmodel: torch.nn.Module, dataloader: torch.utils.data.DataLoader) → float

A function to evaluate model performance with MSE

Parameters:

• predmodel – a regression neural network for cost prediction

• dataloader – Torch dataloader from optDataSet

Returns:

MSE loss

Return type:

float