fix: make rsq regression measure (#1157)

mlr-org · Sep 3, 2024 · d686b87 · d686b87
1 parent 5ffcfee
commit d686b87
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/R/MeasureRegrRSQ.R b/R/MeasureRegrRSQ.R
@@ -29,7 +29,7 @@
 #' @template seealso_measure
 #' @export
 MeasureRegrRSQ = R6Class("MeasureRSQ",
-  inherit = Measure,
+  inherit = MeasureRegr,
   public = list(
     #' @description
     #' Creates a new instance of this [R6][R6::R6Class] class.
@@ -40,10 +40,10 @@ MeasureRegrRSQ = R6Class("MeasureRSQ",
 
       super$initialize(
         id = "rsq",
-        task_type = "regr",
         properties = if (!private$.pred_set_mean) c("requires_task", "requires_train_set") else character(0),
         predict_type = "response",
         minimize = FALSE,
+        range = c(-Inf, 1),
         man = "mlr3::mlr_measures_regr.rsq"
       )
     }

diff --git a/man/mlr_measures_regr.rsq.Rd b/man/mlr_measures_regr.rsq.Rd