H2O 性能指标：AUCPR 不可用？

Question

我正在处理一个二元分类问题，我尝试在 H2O 中使用性能指标 AUCPR，因为我有一个不平衡的数据集。

但是，当我尝试调用 aucpr 指标时出现以下错误。

# entrainement
aml_sans_class_balance.train(x=x_h2o, y=y_h2o, training_frame=train_h2o, validation_frame=valid_h2o)


# Score auc
prediction = aml_sans_class_balance.leader.model_performance(test_data=test_h2o)
tot_auc_score_sans_class_balance.append(prediction.aucpr())
print('score auc à itération', j+1,':',prediction.aucpr())

----------------------------------------------------------------------

AttributeError: type object 'H2OBinomialModelMetrics' has no attribute 'aucpr'

感谢您的帮助。

问候。

Answer 1

你必须调用prediction.pr_auc() 方法而不是prediction.aucpr() 方法。

查看文档： http://docs.h2o.ai/h2o/latest-stable/h2o-py/docs/metrics.html#h2o.model.metrics_base.MetricsBase.pr_auc