make criteria to classify a forecast as binary stricter

epiforecasts · Mar 15, 2021 · ec3fdf5 · ec3fdf5
1 parent 04d85d2
commit ec3fdf5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/eval_forecasts.R b/R/eval_forecasts.R
@@ -332,7 +332,7 @@ eval_forecasts <- function(data = NULL,
   }
 
   if (all.equal(data$true_value, as.integer(data$true_value)) == TRUE) {
-    if (all(data$true_value %in% c(0,1))) {
+    if (all(data$true_value %in% c(0,1)) && all(data$prediction >= 0) && all(data$prediction <= 1)) {
       target_type = "binary"
     } else {
       target_type = "integer"