mindspore.nn.CosineSimilarity
- class mindspore.nn.CosineSimilarity(similarity='cosine', reduction='none', zero_diagonal=True)[source]
Computes representation similarity.
- Parameters
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import numpy as np >>> from mindspore import nn >>> >>> test_data = np.array([[1, 3, 4, 7], [2, 4, 2, 5], [3, 1, 5, 8]]) >>> metric = nn.CosineSimilarity() >>> metric.clear() >>> metric.update(test_data) >>> square_matrix = metric.eval() >>> print(square_matrix) [[0. 0.94025615 0.95162452] [0.94025615 0. 0.86146098] [0.95162452 0.86146098 0.]]
- eval()[source]
Computes the similarity matrix.
- Returns
numpy.ndarray. The similarity matrix.
- Raises
RuntimeError – If the update method is not called first, an error will be reported.
- update(inputs)[source]
Updates the internal evaluation result with ‘inputs’.
- Parameters
inputs (Union[Tensor, list, numpy.ndarray]) – The input matrix.