當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。