javascript - 从 Tensorflow.js 文本毒性检测模型返回 'predictions'

Question

我目前正在尝试在我为大学创建的网站上实现评论部分。
我的目标是检查评论是否有毒，或者是否有警告或发布评论。
我正在使用来自 Tensowflow.js 的预训练模型。
现在我很难返回“预测”数组，因此我可以检查模型是否返回有毒分类。

//Model laden
toxicity.load(grenzwert).then(model => {
    //Kommentar als modelinput
    var saetze = kommentar;
    var modeloutput = new Object();

    //Vorhersage
    model.classify(saetze).then(predictions => {
        modeloutput = predictions;
        return modeloutput;
    });
    //Vorhersage übergeben
    console.log(modeloutput);
});

提前致谢。

Answer 1

您将需要等待预测结果

modeloutput = await model.classify(saetze)

console.log(modeloutput);

唯一要记住的是 await 需要在异步函数中