mean_squared_error
MeanSquaredError
¶
Bases: LossOp
Calculate the mean squared error loss between two tensors.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
inputs
|
Union[Tuple[str, str], List[str]]
|
A tuple or list like: [y_pred, y_true]. |
required |
outputs
|
str
|
String key under which to store the computed loss. |
required |
mode
|
Union[None, str, Iterable[str]]
|
What mode(s) to execute this Op in. For example, "train", "eval", "test", or "infer". To execute regardless of mode, pass None. To execute in all modes except for a particular one, you can pass an argument like "!infer" or "!train". |
'!infer'
|
ds_id
|
Union[None, str, Iterable[str]]
|
What dataset id(s) to execute this Op in. To execute regardless of ds_id, pass None. To execute in all ds_ids except for a particular one, you can pass an argument like "!ds1". |
None
|
average_loss
|
bool
|
Whether to average the element-wise loss after the Loss Op. |
True
|