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