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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。