从 AutoML PredictResponse 对象返回分数

Question

我正在尝试从我的 google automl PredictResponse 对象中提取分数值。从 API 返回的对象如下所示：

payload {

  classification {

    score: 0.989063024521

  }

  display_name: "No"

}

payload {

  classification {

    score: 0.0109369996935

  }

  display_name: "Yes"

}

我已经接近使用这个：

result.payload[1]

返回：

classification {

  score: 0.0109369996935

}

display_name: "Yes"

但我仍然不能只得到是或否的分数。

我试过使用 simplejson，但 PredictResponse 对象也不能很好地使用它。我可以使用对象中的函数来获得是的分数和否的分数吗？感谢帮助！

Answer 1

result.payload[1].classification.score

Answer 2

您可以使用此代码d = {} for i in range(len(response.payload)): d[response.payload[i].display_name] = response.payload[i].classification.score它将为您提供所有类别的字典（在您的情况下是和否）

Answer 3

您可以使用以下方法提取 display_name 字段：result.payload[N].display_name在Vision and Natural Language AutoML 文档中，您可以使用下面的循环来返回您的预测的 display_name。for result in response.payload:    print("Predicted class name: {}".format(result.display_name))