MedianAbsoluteError

class ignite.contrib.metrics.regression.MedianAbsoluteError(output_transform=<function MedianAbsoluteError.<lambda>>, device=device(type='cpu'))

Calculates the Median Absolute Error.

$$\text{MdAE} = \text{MD}_{j=1,n} \left( |A_j - P_j| \right)$$

where $A_j$ is the ground truth and $P_j$ is the predicted value.

More details can be found in Botchkarev 2018.

• update must receive output of the form (y_pred, y) or {'y_pred': y_pred, 'y': y}.

• y and y_pred must be of same shape (N, ) or (N, 1) and of type float32.

Warning

Current implementation stores all input data (output and target) in as tensors before computing a metric. This can potentially lead to a memory error if the input data is larger than available RAM.

Parameters

• output_transform (Callable) – a callable that is used to transform the Engine’s process_function’s output into the form expected by the metric. This can be useful if, for example, you have a multi-output model and you want to compute the metric with respect to one of the outputs. By default, metrics require the output as (y_pred, y) or {'y_pred': y_pred, 'y': y}.

• device (Union[str, device]) – optional device specification for internal storage.

Methods