confusion_matrix() with no parameters does not work

h2oai / h2o-3

H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

Apache License 2.0

6.92k stars 2k forks source link

If I do m.confusion_matrix(), where is a multinomial model, I get:

TypeError: confusion_matrix() takes exactly 2 arguments (1 given)

But m.confusion_matrix(train) works. This is different to the way R works (h2o.confusionMatrix(m)). It is also different to the way binomial models work:

http://docs.h2o.ai/h2o/latest-stable/h2o-py/docs/model_categories.html#h2o.model.multinomial.H2OMultinomialModel.confusion_matrix

http://docs.h2o.ai/h2o/latest-stable/h2o-py/docs/model_categories.html#h2o.model.H2OBinomialModel.confusion_matrix

h2oai / h2o-3

confusion_matrix() with no parameters does not work #10380