Enable set auc_type parameter when using H2OAutoML

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.

http://h2o.ai

Apache License 2.0

6.92k stars 2k forks source link

Enable set auc_type parameter when using H2OAutoML #16184

Open MohanadElMetwally opened 6 months ago

MohanadElMetwally commented 6 months ago

When I train a model using H2OAutoML for a multiclassification problem there is a parameters called auc_type that prevents the calculation of multinomial_auc_table and multinomial_aucpr_table since its set to "auto" by default. How can I change this parameter to make those 2 functions work?

maurever commented 6 months ago

Hi @MohanadElMetwally. Thanks for reporting this issue. This is a bug we need to fix it.

We need to add the possibility of setting auc_type parameter to users of H2OAutoML.