当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python mxnet.metric.np用法及代码示例


用法:

mxnet.metric.np(numpy_feval, name=None, allow_extra_outputs=False)

参数：

  • numpy_feval：(callable(label, pred)) - 自定义评估函数，以 numpy 数组的形式接收小批量的标签和预测，并将相应的自定义指标作为浮点数返回。
  • name：(str, optional) - 自定义指标的名称。
  • allow_extra_outputs：(bool, optional) - 预测输出是否允许有额外的输出。这在像 RNN 这样的情况下很有用，其中状态也是输出的一部分，然后可以在下一步中反馈给 RNN。默认情况下，不允许额外的输出。

返回：

与提供的标签和预测对应的自定义指标。

返回类型：

浮点数

创建一个自定义评估指标，以 numpy 数组的形式接收其输入。

示例：

>>> def custom_metric(label, pred):
...     return np.mean(np.abs(label-pred))
...
>>> metric = mx.metric.np(custom_metric)

相关用法


注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.metric.np。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。