mindspore.train.CosineSimilarity ================================= .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/train/mindspore.train.CosineSimilarity.rst :alt: 查看源文件 .. py:class:: mindspore.train.CosineSimilarity(similarity='cosine', reduction='none', zero_diagonal=True) 计算余弦相似度。 参数: - **similarity** (str) - 计算逻辑。 ``"cosine"`` 表示相似度计算逻辑, ``"dot"`` 表示矩阵点乘计算逻辑。默认值: ``'cosine'`` 。 - **reduction** (str) - 规约计算方式。支持 ``"none"`` 、 ``"sum"`` 或 ``"mean"`` 。默认值: ``'none'`` 。 - **zero_diagonal** (bool) - 如果为 ``True`` ,则对角线将设置为零。默认值: ``True`` 。 .. py:method:: clear() 重置评估结果。 .. py:method:: eval() 计算Cosine Similarity矩阵。 返回: numpy.ndarray,相似度矩阵。 异常: - **RuntimeError** - 如果没有先调用update方法。 .. py:method:: update(*inputs) 使用 `inputs` 更新内部评估结果。 参数: - **inputs** (Union[Tensor, list, numpy.ndarray]) - 输入的矩阵,包含 `y_pred` 和 `y` 。