当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Tensorflow.js tf.losses.hingeLoss()用法及代码示例


Tensorflow.js是Google开发的开源库,用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。

Tensorflow.js tf.losses.hingeLoss() 函数计算两个给定张量之间的铰链损失。

用法:

tf.losses.hingeLoss (labels, predictions, weights, reduction)

参数:

  • labels:它指定了真值输出张量。基于该张量预测绝对差异。
  • predictions:它指定与标签具有相同维度的预测输出张量。
  • weights:它指定一个秩张量,或者等于标签的秩,以便它可以广播,或者为 0。它是一个可选参数。
  • reduction:它指定了减少损失的类型。它是一个可选参数。

返回值:它返回一个由 hingeLoss() 函数计算的 tf.Tensor。



范例1:在这个例子中,我们将采用两个 2d 张量作为标签和预测。然后我们将找到这两个张量之间的估计铰链损失。

Javascript


// Importing the tensorflow.js library 
const tf = require("@tensorflow/tfjs"); 
  
// Defining label tensor 
const x_label = tf.tensor2d([ 
    [0., 1., 0.],  
    [1., 0., 1.] 
]); 
  
// Defining prediction tensor 
const x_pred = tf.tensor2d([ 
    [1., 1., 1.],  
    [0., 0., 0. ] 
]); 
  
// Calculating hinge loss
const hinge_loss = tf.losses.hingeLoss(x_label,x_pred) 
    
// Printing the output 
hinge_loss.print()

输出:

Tensor
    1.1666667461395264

范例2:

Javascript


// Importing the tensorflow.Js lbrary
import * as tf from "@tensorflow/tfjs"
  
// Computing hinge loss between two 
// tensors and printing the result
tf.losses.hingeLoss(
    tf.tensor4d([[[[0], [4]], [[5], [1]]]]), 
    tf.tensor4d([[[[1], [2]], [[3], [4]]]])
).print();

输出:

Tensor
    0.5

参考: https://js.tensorflow.org/api/1.0.0/#losses.hingeLoss




相关用法


注:本文由纯净天空筛选整理自abhinavjain194大神的英文原创作品 Tensorflow.js tf.losses.hingeLoss() Function。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。