FIX: hyperband now supported with stacking

civisanalytics · Aug 2, 2018 · e5320d9 · e5320d9
1 parent a9cc657
commit e5320d9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,11 @@
 # Changelog
 
+## [1.5.1] - Unreleased
+
+### Changed
+
+- Hyperband is now supported for stacking estimators in `civis_ml`. Fixes #131.
+
 ## [1.5.0] - 2018-05-20
 
 ### Changed

diff --git a/R/civis_ml.R b/R/civis_ml.R
@@ -590,8 +590,7 @@ create_and_run_model <- function(file_id = NULL,
 
   if (!is.null(cross_validation_parameters)) {
     hyperband <- identical(cross_validation_parameters, "hyperband")
-    hyperband_not_supported <- model_type %in% c("sparse_logistic", "sparse_linear_regressor",
-                                                 "stacking_regressor")
+    hyperband_not_supported <- model_type %in% c("sparse_logistic", "sparse_linear_regressor")
     if (hyperband & hyperband_not_supported) {
       stop(paste0("cross_validation_parameters = \"hyperband\" not supported for ", model_type))
     }