machine-learning - h2o 如何返回 mcc 值

Question

我创建随机森林模型。 mcc 值是两个值的列表。为什么？

mRF=H2ORandomForestEstimator(nfolds=10,keep_cross_validation_models = True,seed=12345,model_id="RF0",ntrees=1000)
mRF.train(x=column_use, y=target, training_frame=train,validation_frame=valid)
print("MCC valid",mRF.model_performance(valid).mcc())
MCC valid [[0.35743618321170406, 0.21239407659849494]]

Answer 1

如果您不设置参数阈值，它将返回一个数组，其中包含最佳阈值的一个结果 - [[最佳阈值，mcc]]。您可以设置自己的阈值数组 - [threshold1, Threshold2, ...] 它将返回 [[threshold1, mcc1], [threshold2, mcc2], ... ]。