从Tensorflow.js文本毒性检测模型返回“预测”

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

您需要await预测结果

modeloutput = await model.classify(saetze)

console.log(modeloutput);

唯一需要记住的是await必须处于异步函数中