我在处理一个二分类问题,由于数据集不平衡,我尝试在H2O中使用性能指标AUCPR。
然而,当我尝试调用aucpr指标时,收到了以下错误。
# entrainementaml_sans_class_balance.train(x=x_h2o, y=y_h2o, training_frame=train_h2o, validation_frame=valid_h2o)# Score aucprediction = 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'
感谢您的帮助。
此致。
回答:
您需要调用prediction.pr_auc()
方法,而不是prediction.aucpr()
方法。