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


Python mxnet.metric.create用法及代碼示例

用法:

mxnet.metric.create(metric, *args, **kwargs)

參數

  • metric(str or callable) -指定要創建的指標。此參數必須是以下之一:
    • 度量的名稱。
    • EvalMetric 的一個實例。
    • 一個列表,其中的每個元素都是一個度量或度量名稱。
    • 計算給定批次標簽和預測的自定義指標的評估函數。
  • *args(list) - 度量構造函數的附加參數。僅在 metric 為 str 時使用。
  • **kwargs(dict) - 度量構造函數的附加參數。僅在 metric 為 str 時使用

從指標名稱或EvalMetric 實例或自定義指標函數創建評估指標。

例子

>>> def custom_metric(label, pred):
...     return np.mean(np.abs(label - pred))
...
>>> metric1 = mx.metric.create('acc')
>>> metric2 = mx.metric.create(custom_metric)
>>> metric3 = mx.metric.create([metric1, metric2, 'rmse'])

相關用法


注:本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.metric.create。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。