Merge branch 'main' of https://github.com/EOGrady21/vprr

EOGrady21 · Dec 12, 2024 · 2b14b85 · 2b14b85
2 parents e794bd2 + c76d167
commit 2b14b85
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/R/EC_functions.R b/R/EC_functions.R
@@ -31,9 +31,19 @@ read_aid_cnn <- function(aid_file) {
   #' @export
   #'
   aid_table <- read.table(aid_file, sep = " ")
-  names(aid_table) <- c('roi', 'confidence')
+
+  if(ncol(aid_table) == 1) {
+
+    names(aid_table) <- c('roi')
+
+  } else {
+
+    names(aid_table) <- c('roi', 'confidence')
+
+  }
 
   return(aid_table)
+
 }
 
 vpr_pred_read <- function(filename) {