h20无人驾驶AI ROC曲线如何识别多类混淆矩阵的阈值

时间:2018-12-14 05:44:26

标签: driverless-ai

我创建了一个训练模型,并且ROC曲线在Multiclass Confusion Matrix上显示0.9748。我使用“在另一个数据集上的得分”在测试数据上运行了该模型,并得到了预测。我想了解如何获得这些预测的阈值,以便我们可以将将来的值发布给用户。

1 个答案:

答案 0 :(得分:0)

DAI返回的预测值不是标签。这意味着您必须自己设置阈值。例如,您可以下载预测文件,然后将其导入您喜欢的语言(例如,使用H2O-3的python api),然后运行布尔检查以查看给定列的值是否高于其阈值成为特定标签。

有关多类实验图的详细信息以及DAI如何决定显示不同的阈值指标,请参见文档here