mindspore/docs/api/api_python/train/mindspore.train.Dice.rst

40 lines
1.3 KiB
ReStructuredText
Raw Normal View History

mindspore.train.Dice
=====================
2022-02-15 10:53:07 +08:00
.. py:class:: mindspore.train.Dice(smooth=1e-5)
2022-02-15 10:53:07 +08:00
集合相似性度量。
2022-03-28 17:31:19 +08:00
用于计算两个样本之间的相似性。当分割结果最好时Dice系数的值为1当分割结果最差时Dice系数的值为0。Dice系数表示预测值与真实值交集同预测值和真实值并集之间的比值。
2022-02-15 10:53:07 +08:00
.. math::
dice = \frac{2 * (pred \bigcap true)}{pred \bigcup true}
2022-07-22 16:25:38 +08:00
参数:
- **smooth** (float) - 在计算过程中添加到分母里用于提高数值稳定性取值需大于0。默认值1e-5。
2022-02-15 10:53:07 +08:00
.. py:method:: clear()
重置评估结果。
.. py:method:: eval()
2022-03-28 17:31:19 +08:00
计算Dice系数。
2022-02-15 10:53:07 +08:00
2022-07-22 16:25:38 +08:00
返回:
Float计算的结果。
2022-02-15 10:53:07 +08:00
2022-07-22 16:25:38 +08:00
异常:
- **RuntimeError** - 样本数为0。
2022-02-15 10:53:07 +08:00
.. py:method:: update(*inputs)
更新内部评估结果 `y_pred``y`
2022-07-22 16:25:38 +08:00
参数:
2022-09-27 15:19:36 +08:00
- **inputs** (tuple) - 输入 `y_pred``y``y_pred``y` 是tensor、列表或numpy.ndarray。 `y_pred` 是预测值, `y` 是真实值。 `y_pred``y` 的shape都是 `(N, ...)`
2022-02-15 10:53:07 +08:00
2022-07-22 16:25:38 +08:00
异常:
- **ValueError** - 输入参数的数量不等于2。
- **ValueError** - 如果预测值和标签shape不一致。