我正在编写一个自定义损失函数,该函数用作惩罚矩阵来计算损失。该y_true
和y_pred
值是矩阵罚矩阵的指标A
:
def score(y_true, y_pred):
S = 0.0
for i in range(0, y_true.shape[0]):
S -= A[y_true[i], y_pred[i]]
return S/y_true.shape[0]
由于for循环无法提供最佳性能,因此我想知道如果没有它,如何执行以下代码段。
S -= A[y_true[i], y_pred[i]]
在我看来,如果y_true
包含行索引和y_pred
列索引,以下内容将为您服务
S = -sum(A[y_true,y_pred])
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句