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

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

.confusionMatrix() 函數用於根據所述真實標簽和預測標簽計算混淆矩陣。

用法:

tf.confusionMatrix(labels, predictions, numClasses)

Parameters:

labels:聲明的目標標簽應該是基於零的整數，以支持類。它有形狀 [numExamples]。其中，numExamples 是合並實例的度量。它可以是 tf.Tensor1D、TypedArray 或數組類型。
predictions:聲明的預測類應該是基於零的整數，以支持類。它的形狀應與所述標簽相同。它可以是 tf.Tensor1D、TypedArray 或數組類型。
numClasses:它是整數類型的總類數。此外，其度量應大於所述標簽和預測中的最大元素。它是類型號。

返回值：它返回 tf.Tensor2D 對象。

範例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining predictions, labels and 
// numClasses
const lab = tf.tensor1d([3, 4, 1, 0, 1], 'int32');
const pred = tf.tensor1d([1, 3, 0, 4, 1], 'int32');
const num_Cls = 2;
  
// Calling tf.confusionMatrix() method
const output = tf.math.confusionMatrix(lab, pred, num_Cls);
  
// Printing output
output.print();

輸出：

Tensor
    [[0, 0],
     [1, 1]]

範例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling tf.confusionMatrix() method
const res = tf.math.confusionMatrix(
    tf.tensor1d([3.3, 4.5, null, 'a', 'b']), 
    tf.tensor1d([-2, 5.3, -0.1, 4.3, 12.5]), 4
);
  
// Printing output
res.print();

輸出：

Tensor
    [[1, 0, 0, 0],
     [0, 0, 0, 0],
     [0, 0, 0, 0],
     [0, 0, 0, 0]]

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

相關用法

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