【发布时间】:2021-02-11 15:42:08
【问题描述】:
我正在使用 H2ORandomForestEstimator 进行多类分类。
构建和训练后如下:
train, valid = hdf.split_frame(ratios=[.8], seed=1234)
# Build and train the model:
drf = H2ORandomForestEstimator(model_id="drf", seed=1234)
drf.train(x=predictors,
y=response,
training_frame=train,
validation_frame=valid)
drf.model_performance(valid)
我可以在输出中看到每个类的 RMSE、MSE 和平均误差
ModelMetricsMultinomial: drf
** Reported on test data. **
MSE: 0.12204577776460168
RMSE: 0.34935050846478194
LogLoss: 0.4781165975023516
Mean Per-Class Error: 0.23864386780117242
我如何获得其他指标,例如准确度、精确度、召回率和 F-Score?
【问题讨论】:
-
请包含一个完全(非部分)可重现的示例。谢谢! stackoverflow.com/help/minimal-reproducible-example
标签: python random-forest h2o