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

Question

我目前正在尝试在我为大学创建的网站上实现评论部分。
我的目标是检查评论是否有毒，是否会发出警报或发布评论。
我正在使用来自 Tensowflow.js.
的预训练模型现在我努力返回 'predictions' 数组，所以我可以检查模型 returns 是否属于有毒分类。

//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需要在异步函数中

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

Returning the 'predictions' from Tensorflow.js Text toxicity detection model

javascript

arrays

json

tensorflow

tensorflow.js