Browse Source

update the documentation of F1 operator.

tags/v1.2.0-rc1
wangshuide2020 5 years ago
parent
commit
3e481f85b9
2 changed files with 4 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/metrics/fbeta.py
  2. +3
    -0
      mindspore/ops/operations/nn_ops.py

+ 1
- 1
mindspore/nn/metrics/fbeta.py View File

@@ -122,7 +122,7 @@ class Fbeta(Metric):
class F1(Fbeta):
r"""
Calculates the F1 score. F1 is a special case of Fbeta when beta is 1.
Refer to class :class: `mindspore.nn.Fbeta` for more details.
Refer to class :class:`mindspore.nn.Fbeta` for more details.

.. math::
F_1=\frac{2\cdot true\_positive}{2\cdot true\_positive + false\_negative + false\_positive}


+ 3
- 0
mindspore/ops/operations/nn_ops.py View File

@@ -5765,10 +5765,13 @@ class ApplyFtrl(PrimitiveWithInfer):

Outputs:
There are three outputs for Ascend environment.

- **var** (Tensor) - represents the updated `var`.
- **accum** (Tensor) - represents the updated `accum`.
- **linear** (Tensor) - represents the updated `linear`.

There is only one output for GPU environment.

- **var** (Tensor) - This value is alwalys zero and the input parameters has been updated in-place.

Supported Platforms:


Loading…
Cancel
Save