predictions should be a numeric

OHDSI · Dec 5, 2023 · 02d3d44 · 02d3d44
1 parent 46180e4
commit 02d3d44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/Estimator.R b/R/Estimator.R
@@ -267,7 +267,7 @@ predictDeepEstimator <- function(plpModel,
   } else {
     prediction$value <- plpModel$model$predict_proba(data)
   }
-
+  prediction$value <- as.numeric(prediction$value)
   attr(prediction, "metaData")$modelType <- attr(plpModel, "modelType")
 
   return(prediction)