如何从节点的className中获取最可能的TensorFlow.js

Question

我想要做的事情：，我想从mobileNet的TensorFlow.js图像分类中得到概率最高的className。我想以字符串的形式得到className。

问题：--我不知道如何将className隔离为字符串。这有命令吗？或者你知道我怎么解决我的问题吗？

我的代码：

theAi();

async function theAi() {
    const tf = require('@tensorflow/tfjs'),
    mobilenet = require('@tensorflow-models/mobilenet'),
    tfnode = require('@tensorflow/tfjs-node'),
    fs = require('fs-extra');

    const imageBuffer = await fs.readFile("./jesus.jpg"),
    tfimage = tfnode.node.decodeImage(imageBuffer),
    mobilenetModel = await mobilenet.load();  

    const results = await mobilenetModel.classify(tfimage);
    console.log(results);
};

和输出：

[
  {
    className: 'chain mail, ring mail, mail, chain armor, chain armour, ring armor, ring armour',
    probability: 0.5979475378990173
  },
  { className: 'vestment', probability: 0.14261281490325928 },
  { className: 'fountain', probability: 0.03441018983721733 }
]

所以在这种情况下，我想得到字符串(‘链邮件，戒指邮件，邮件，链盔甲，链甲，环甲，环甲’)。

Answer 1

这是一个纯粹的javascript问题。如果得到这个结果，只需搜索m aximum：

const maxClass = result.reduce(function(prev, current) {
    return (prev.probability > current.probability) ? prev : current;
})["className"];