如何保存与NLP API预测相关的分数?

时间:2019-12-02 15:03:49

标签: google-cloud-platform google-cloud-firestore google-cloud-functions google-cloud-nl

因此,目前,我的扩展程序通过云功能将一段文本发送到NLP API。这段文本经过处理和预测,并根据预测分配分数(例如,句子可以为0.33-“重要同意”)。我想知道是否可以将带有相应分数的句子保存到Firestore中。目前,我只能保存句子,而不能保存分数。

由于我们使用的阈值限制,我们真的很想在Firestore数据库中获得分数。如果没有分数,则阈值将过时。

这里是云功能,以防万一:

  exports.queryAutoML = (req, res) => {

  const automl = require('@google-cloud/automl');

  const client = new automl.PredictionServiceClient();

  var formattedName = client.modelPath('*********', '**********', '*****************');
  var payload = {
    "textSnippet": {
       "content": req.body,
        "mime_type": "text/plain"
    },
  };
  var request = {
    name: formattedName,
    payload: payload,
  };
  client.predict(request)
    .then(responses => {
    console.log("in success");
    let title = responses[0].payload[0].displayName;
    let score = responses[0].payload[0].classification.score;
    output = [req.body, title, score];
    res.status(200).send(output);
  })
    .catch(err => {
    console.log("in error");
    console.error(err);
  });

1 个答案:

答案 0 :(得分:0)

将Frank van Puffelen的答案添加为社区Wiki,以提高可见性:

评论link中提供的摘录显示了如何创建文档并将其上传到Firestore集合。因此,您需要做的就是使用NLP查询的结果作为字段来创建文档,然后使用Firebase SDK对其进行推送。