fix: dependencies in default svm configuration

mlr-org · Feb 20, 2024 · a888203 · a888203
1 parent 2dc91de
commit a888203
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/R/LearnerClassifSVM.R b/R/LearnerClassifSVM.R
@@ -94,8 +94,14 @@ default_values.LearnerClassifSVM = function(x, search_space, task, ...) { # noli
     gamma = 1 / length(task$feature_names)
   )
   defaults = insert_named(default_values(x$param_set), special_defaults)
-  defaults[["degree"]] = NULL
-  defaults[search_space$ids()]
+  # defaults[["degree"]] = NULL
+  defaults = defaults[search_space$ids()]
+
+  # fix dependencies
+  if (!is.null(defaults[["degree"]])) defaults[["degree"]] = NA_real_
+  if (!is.null(defaults[["coef0"]])) defaults[["coef0"]] = NA_real_
+
+  defaults
 }
 
 #' @include aaa.R