Skip to content

MSE

Description

Calculating the error and the gradient on the batch: (mean squared error - MSE), which is the average sum of the squared differences between the network responses and the real labels.

It is used in regression tasks.

The error function formula is:

MSE = \frac{1}{N}\sum_{i=1}^N(y_i-y_i^p)^2

where

N - number of objects in the sample;
y_i - real value for the i-th object;
y_i^p - predicted value for the i-th object.

Initializing

def __init__(self):

Parametrs

-

Explanations

-

Examples


Necessary imports:

import numpy as np
from PuzzleLib.Backend import gpuarray
from PuzzleLib.Cost import MSE

Info

gpuarray is required to properly place the tensor in the GPU.

is required to properly place the tensor in the GPU.

targets = gpuarray.to_gpu(np.random.randn(10, 10).astype(np.float32))
predictions = gpuarray.to_gpu(np.random.randn(10, 10).astype(np.float32))

Important

Please remember that the first dimension of tensors is the size of the batch

Initializing the error function:

mse = MSE()

Calculating the error and the gradient on the batch:

error, grad = mse(predictions, targets)