如何获得h2o模型的准确度/精度?

时间:2016-08-02 01:18:37

标签: python machine-learning classification h2o

我尝试使用逻辑回归来获得多类分类器的准确性。有没有办法通过内置函数获得准确性,还是我必须自己编写函数?

到目前为止我的代码下面

multinomial_fit = H2OGeneralizedLinearEstimator(family="multinomial",max_iterations=100)

multinomial_fit.train(x=train_h2o_cro.columns[1:],y=train_h2o_cro.columns[0],training_frame=train_h2o)

prediction_glm_h2o = multinomial_fit.predict(test_h2o)

multinomial_fit.model_performance(test_h2o)

使用最后一行代码,我只得到mse而没有别的。

提前致谢。

2 个答案:

答案 0 :(得分:3)

目前尚未实现,但添加此功能是有意义的。以下是JIRA ticket,您可以在其中跟踪进度。

答案 1 :(得分:-2)

<强>被修改

由于Erin指出目前没有实现准确性,因此评估模型性能的选项仅限于H2OMultinomialModelMetrics可用的功能。

例如,您可以查看.mean_per_class_error(),查看多类混淆矩阵model.confusion_matrix(data)或日志丢失.logloss()等等。