Tensorflow.js tf.inTopKAsync()用法及代碼示例

Tensorflow.js是由Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型以及深度學習神經網絡。

.inTopKAsync()函數用於檢查指定的目標是否在給定的前K個預測中。

用法：

tf.inTopKAsync(predictions, targets, k?)

Parameters:

predictions:它是指定的二維或上張量輸入，其最後大小不小於k，並且可以是tf.Tensor，TypedArray或Array類型。
targets:它是指定的一維或上張量輸入，可以是tf.Tensor，TypedArray或Array類型。
k:這是計算精度時要考慮的頂級元素的替代數量。默認值是1，它是數字類型。

返回值：它返回Promise tf.Tensor對象。

範例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining predictions and targets
const pred = tf.tensor2d([
    [11, 22, 33, 55], 
    [33, 66, 22, -11]
]);
  
const targ = tf.tensor1d([1, 1]);
  
// Calling tf.inTopKAsync() method
const res = await tf.inTopKAsync(pred, targ);
  
// Printing output
res.print();

輸出：

Tensor
    [false, true]

範例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling tf.inTopKAsync() method with 
// all its parameters
const res = await tf.inTopKAsync(
    tf.tensor2d([[1.1, 2.2], [3.3, 6.6]]), 
    tf.tensor1d([0, 1]), 2);
  
// Printing output
res.print();

輸出：

Tensor
    [true, true]

參考： https://js.tensorflow.org/api/latest/#inTopKAsync

相關用法

注：本文由純淨天空篩選整理自nidhi1352singh大神的英文原創作品 Tensorflow.js tf.inTopKAsync() Function。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。