mindspore.train.Fbeta

查看源文件
class mindspore.train.Fbeta(beta)[源代码]

计算Fbeta评分。

Fbeta评分是精度(Precision)和召回率(Recall)的加权平均值。

Fβ=(1+β2)true_positive(1+β2)true_positive+β2false_negative+false_positive
参数:
  • beta (Union[float, int]) - F-measure中的beta系数。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindspore.train import Fbeta
>>>
>>> x = Tensor(np.array([[0.2, 0.5], [0.3, 0.1], [0.9, 0.6]]))
>>> y = Tensor(np.array([1, 0, 1]))
>>> metric = Fbeta(1)
>>> metric.clear()
>>> metric.update(x, y)
>>> fbeta = metric.eval()
>>> print(fbeta)
[0.66666667 0.66666667]
clear()[源代码]

内部评估结果清零。

eval(average=False)[源代码]

计算Fbeta结果。

参数:
  • average (bool) - 是否计算Fbeta平均值。默认值: False

返回:

numpy.ndarray或numpy.float64,计算的Fbeta评分结果。

update(*inputs)[源代码]

使用预测值 y_pred 和真实标签 y 更新内部评估结果。

参数:
  • inputs - y_predyy_predy 支持Tensor、list或numpy.ndarray类型。 通常情况下, y_pred[0,1] 之间的浮点数列表,shape为 (N,C) ,其中 N 是样本数, C 是类别数。 y 是整数值,如果使用one-hot编码,则shape为 (N,C) ;如果使用类别索引,shape是 (N,)

异常:
  • ValueError - 当前输入的 y_pred 和历史 y_pred 类别数不匹配。

  • ValueError - 预测值和真实值包含的类别不同。